diff --git a/tools/configs/r50_fpn_frcnn_1x.py b/tools/configs/r50_fpn_frcnn_1x.py index 6ab3dbc36173d19ff74e9b0cfcdab2d8d045f342..09167dd09b18a5946766afaf1d5a96277fd6ceae 100644 --- a/tools/configs/r50_fpn_frcnn_1x.py +++ b/tools/configs/r50_fpn_frcnn_1x.py @@ -131,7 +131,7 @@ lr_config = dict( checkpoint_config = dict(interval=1) # yapf:disable log_config = dict( - interval=20, + interval=50, hooks=[ dict(type='TextLoggerHook'), # dict(type='TensorboardLoggerHook', log_dir=work_dir + '/log') diff --git a/tools/configs/r50_fpn_maskrcnn_1x.py b/tools/configs/r50_fpn_maskrcnn_1x.py index 677176c56b642bda0dff3486185687e027a336c9..35dab5633c95b855f4d32a8245aaf44c69cddced 100644 --- a/tools/configs/r50_fpn_maskrcnn_1x.py +++ b/tools/configs/r50_fpn_maskrcnn_1x.py @@ -144,10 +144,10 @@ lr_config = dict( checkpoint_config = dict(interval=1) # yapf:disable log_config = dict( - interval=20, + interval=50, hooks=[ dict(type='TextLoggerHook'), - # ('TensorboardLoggerHook', dict(log_dir=work_dir + '/log')), + # dict(type='TensorboardLoggerHook', log_dir=work_dir + '/log') ]) # yapf:enable # runtime settings diff --git a/tools/train.py b/tools/train.py index 1c573515acb594f074f4b4f2467aa6b7b32e8321..f60b5c0bffd3c602eca75cc0d03eb9eb383dd951 100644 --- a/tools/train.py +++ b/tools/train.py @@ -90,8 +90,8 @@ def main(): cfg.work_dir = args.work_dir cfg.gpus = args.gpus # add mmdet version to checkpoint as meta data - cfg.checkpoint_config.meta = dict(mmdet_version=__version__) - cfg.checkpoint_config.config = cfg.text + cfg.checkpoint_config.meta = dict( + mmdet_version=__version__, config=cfg.text) logger = get_logger(cfg.log_level)