From c20c2c8812e59e4c3113ae4e20f5ce212dc3b276 Mon Sep 17 00:00:00 2001
From: pangjm <pjmzju@gmail.com>
Date: Mon, 1 Oct 2018 22:25:53 +0800
Subject: [PATCH] minor modify
---
tools/train.py | 12 ++++++++----
1 file changed, 8 insertions(+), 4 deletions(-)
diff --git a/tools/train.py b/tools/train.py
index 6ca0464..2aa681a 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -116,6 +116,11 @@ def main():
build_dataloader(train_dataset, cfg.data.imgs_per_gpu,
cfg.data.workers_per_gpu, cfg.gpus, dist)
]
+ if args.validate:
+ val_dataset = obj_from_dict(cfg.data.test, datasets)
+ data_loaders.append(
+ build_dataloader(val_dataset, cfg.data.imgs_per_gpu,
+ cfg.data.workers_per_gpu, cfg.gpus, dist))
# build model
model = build_detector(
@@ -131,10 +136,9 @@ def main():
if args.validate:
val_dataset = obj_from_dict(cfg.data.test, datasets)
- runner.register_hook(CocoDistEvalmAPHook(val_dataset))
- # data_loaders.append(
- # build_dataloader(val_dataset, cfg.data.imgs_per_gpu,
- # cfg.data.workers_per_gpu, cfg.gpus, dist))
+ data_loaders.append(
+ build_dataloader(val_dataset, cfg.data.imgs_per_gpu,
+ cfg.data.workers_per_gpu, cfg.gpus, dist))
# register hooks
optimizer_config = DistOptimizerHook(
--
GitLab