From fdb443053a232caca99b3f5e70052f0107506afe Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 7 Oct 2018 14:19:00 +0800
Subject: [PATCH] minor fix
---
tools/configs/r50_fpn_frcnn_1x.py | 2 +-
tools/configs/r50_fpn_maskrcnn_1x.py | 4 ++--
tools/train.py | 4 ++--
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/tools/configs/r50_fpn_frcnn_1x.py b/tools/configs/r50_fpn_frcnn_1x.py
index 6ab3dbc..09167dd 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 677176c..35dab56 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 1c57351..f60b5c0 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)
--
GitLab