Method save_config

trainer/base_trainer.py:40–43

(self, config: DictConfig)

        pass

    def save_config(self, config: DictConfig):
        out_path = os.path.join(self.save_dir, 'config.yaml')
        if self.rank == 0:
            OmegaConf.save(config, out_path)

    def setup_save(self, save_dir: str, log_level: str, rank: int = 0, mode='train'):
        log_level_str = log_level.upper()