Skip to content

GitLab

Unverified Commit 097b66ee authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

[Fix] Fix a bug when adding ('val', 1) to workflow (#370) 

* fix a bug when adding val to workflow

* update
parent bc97d76a
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 1 deletion
tools/train.py
View file @ 097b66ee
...@@ -179,7 +179,15 @@ def main(): ...@@ -179,7 +179,15 @@ def main():
datasets = [build_dataset(cfg.data.train)] datasets = [build_dataset(cfg.data.train)]
if len(cfg.workflow) == 2: if len(cfg.workflow) == 2:
val_dataset = copy.deepcopy(cfg.data.val) val_dataset = copy.deepcopy(cfg.data.val)
# in case we use a dataset wrapper
if 'dataset' in cfg.data.train:
val_dataset.pipeline = cfg.data.train.dataset.pipeline
else:
val_dataset.pipeline = cfg.data.train.pipeline val_dataset.pipeline = cfg.data.train.pipeline
# set test_mode=False here in deep copied config
# which do not affect AP/AR calculation later
# refer to https://mmdetection3d.readthedocs.io/en/latest/tutorials/customize_runtime.html#customize-workflow # noqa
val_dataset.test_mode = False
datasets.append(build_dataset(val_dataset)) datasets.append(build_dataset(val_dataset))
if cfg.checkpoint_config is not None: if cfg.checkpoint_config is not None:
# save mmdet version, config file content and class names in # save mmdet version, config file content and class names in
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment