Skip to content

GitLab

Commit c20c2c88 authored by pangjm's avatar pangjm
Browse files

minor modify

parent 070c1ea7
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 4 deletions
tools/train.py
View file @ c20c2c88
...@@ -116,6 +116,11 @@ def main(): ...@@ -116,6 +116,11 @@ def main():
build_dataloader(train_dataset, cfg.data.imgs_per_gpu, build_dataloader(train_dataset, cfg.data.imgs_per_gpu,
cfg.data.workers_per_gpu, cfg.gpus, dist) 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 # build model
model = build_detector( model = build_detector(
...@@ -131,10 +136,9 @@ def main(): ...@@ -131,10 +136,9 @@ def main():
if args.validate: if args.validate:
val_dataset = obj_from_dict(cfg.data.test, datasets) val_dataset = obj_from_dict(cfg.data.test, datasets)
runner.register_hook(CocoDistEvalmAPHook(val_dataset)) data_loaders.append(
# data_loaders.append( build_dataloader(val_dataset, cfg.data.imgs_per_gpu,
# build_dataloader(val_dataset, cfg.data.imgs_per_gpu, cfg.data.workers_per_gpu, cfg.gpus, dist))
# cfg.data.workers_per_gpu, cfg.gpus, dist))
# register hooks # register hooks
optimizer_config = DistOptimizerHook( optimizer_config = DistOptimizerHook(
......
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