"vscode:/vscode.git/clone" did not exist on "5704613e28a742c86bb4fc883534c6c456c4a1b5"
Commit 2d34cebe authored by pangjm's avatar pangjm
Browse files

validate api transfer to hook

parent c20c2c88
...@@ -134,12 +134,6 @@ def main(): ...@@ -134,12 +134,6 @@ def main():
runner = Runner(model, batch_processor, cfg.optimizer, cfg.work_dir, runner = Runner(model, batch_processor, cfg.optimizer, cfg.work_dir,
cfg.log_level) 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 # register hooks
optimizer_config = DistOptimizerHook( optimizer_config = DistOptimizerHook(
**cfg.optimizer_config) if dist else cfg.optimizer_config **cfg.optimizer_config) if dist else cfg.optimizer_config
...@@ -148,6 +142,7 @@ def main(): ...@@ -148,6 +142,7 @@ def main():
if dist: if dist:
runner.register_hook(DistSamplerSeedHook()) runner.register_hook(DistSamplerSeedHook())
# register eval hooks # register eval hooks
if args.validate:
if isinstance(model.module, RPN): if isinstance(model.module, RPN):
runner.register_hook(CocoDistEvalRecallHook(cfg.data.val)) runner.register_hook(CocoDistEvalRecallHook(cfg.data.val))
elif cfg.data.val.type == 'CocoDataset': elif cfg.data.val.type == 'CocoDataset':
......
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