def save_config(self):
    cfg_file = os.path.join(self.ckpt_foldername, 'config.yaml')
    with open(cfg_file, 'w') as f:
        self.config.pop('config_override', None)
        f.write(str(self.config))
def save_config(self):
    cfg_file = os.path.join(self.ckpt_foldername, 'config.yaml')
    with open(cfg_file, 'w') as f:
        self.config.pop('config_override', None)
        f.write(str(self.config))