Add config to runner meta (#3534)

* Add config and config filename to runner meta

* Add absolute path of config file

* change from realpath to abspath

* Use copy.deepcopy

* Use config instead of config_dict.
pull/3649/head
Wang Xinjiang 5 years ago committed by GitHub
parent e76cdfb04b
commit f93c00fd05
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 3
      tools/train.py

@ -130,7 +130,7 @@ def main():
logger.info('Environment info:\n' + dash_line + env_info + '\n' +
dash_line)
meta['env_info'] = env_info
meta['config'] = cfg.pretty_text
# log some basic info
logger.info(f'Distributed training: {distributed}')
logger.info(f'Config:\n{cfg.pretty_text}')
@ -156,7 +156,6 @@ def main():
# checkpoints as meta data
cfg.checkpoint_config.meta = dict(
mmdet_version=__version__ + get_git_hash()[:7],
config=cfg.pretty_text,
CLASSES=datasets[0].CLASSES)
# add an attribute for visualization convenience
model.CLASSES = datasets[0].CLASSES

Loading…
Cancel
Save