Skip to content

Commit

Permalink
fix workflow problem (open-mmlab#2103)
Browse files Browse the repository at this point in the history
* fix workflow problem

* import sort

* update pipeline
  • Loading branch information
abcxs authored Feb 17, 2020
1 parent e530748 commit 3ce959b
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tools/train.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from __future__ import division
import argparse
import copy
import os
import os.path as osp
import time
Expand Down Expand Up @@ -115,7 +116,9 @@ def main():

datasets = [build_dataset(cfg.data.train)]
if len(cfg.workflow) == 2:
datasets.append(build_dataset(cfg.data.val))
val_dataset = copy.deepcopy(cfg.data.val)
val_dataset.pipeline = cfg.data.train.pipeline
datasets.append(build_dataset(val_dataset))
if cfg.checkpoint_config is not None:
# save mmdet version, config file content and class names in
# checkpoints as meta data
Expand Down

0 comments on commit 3ce959b

Please sign in to comment.