INNER CODE UNIT · Python

before_instantiate_classes

Linketic/CityGaussian · internal/cli.py:72

    def before_instantiate_classes(self) -> None:
        config = getattr(self.config, self.config.subcommand)
        if config.name is None:
            # auto set experiment name base on --data.path
            config.name = "_".join(config.data.path.strip("/").split("/")[-3:])
            print("auto determine experiment name: {}".format(config.name))

        if config.max_steps is not None:
            config.trainer.max_steps = config.max_steps
        if config.max_epochs is not None:
            config.trainer.max_epochs = config.max_epochs

        # build output path
        output_path = os.path.join(config.output, config.name)
        if config.version is not None:
            output_path = os.path.join(output_path, config.version)
        if "block_id" in config.data.parser.init_args and config.data.parser.init_args.block_id is not None:
            output_path = os.path.join(output_path, "blocks", "block_{}".format(config.data.parser.init_args.block_id))

View source record →