From 2d34cebe632991745b18466d1032d3592d69f886 Mon Sep 17 00:00:00 2001
From: pangjm <pjmzju@gmail.com>
Date: Mon, 1 Oct 2018 22:29:00 +0800
Subject: [PATCH] validate api transfer to hook
---
tools/train.py | 15 +++++----------
1 file changed, 5 insertions(+), 10 deletions(-)
diff --git a/tools/train.py b/tools/train.py
index 2aa681a..3241174 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -134,12 +134,6 @@ def main():
runner = Runner(model, batch_processor, cfg.optimizer, cfg.work_dir,
cfg.log_level)
- 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))
-
# register hooks
optimizer_config = DistOptimizerHook(
**cfg.optimizer_config) if dist else cfg.optimizer_config
@@ -148,10 +142,11 @@ def main():
if dist:
runner.register_hook(DistSamplerSeedHook())
# register eval hooks
- if isinstance(model.module, RPN):
- runner.register_hook(CocoDistEvalRecallHook(cfg.data.val))
- elif cfg.data.val.type == 'CocoDataset':
- runner.register_hook(CocoDistEvalmAPHook(cfg.data.val))
+ if args.validate:
+ if isinstance(model.module, RPN):
+ runner.register_hook(CocoDistEvalRecallHook(cfg.data.val))
+ elif cfg.data.val.type == 'CocoDataset':
+ runner.register_hook(CocoDistEvalmAPHook(cfg.data.val))
if cfg.resume_from:
runner.resume(cfg.resume_from)
--
GitLab