Skip to content
Snippets Groups Projects
Commit 8002b13f authored by Neta Zmora's avatar Neta Zmora
Browse files

object detection: fix model summary generation

When multi-processing, we want only one process to generate the
summary, while the other processes do nothing (lazy bums!)
parent 801a26a8
No related branches found
No related tags found
No related merge requests found
......@@ -145,9 +145,10 @@ def main(args):
patch_fastrcnn(model)
model.to(device)
if args.summary and utils.is_main_process():
for summary in args.summary:
distiller.model_summary(model, summary, args.dataset)
if args.summary:
if utils.is_main_process():
for summary in args.summary:
distiller.model_summary(model, summary, args.dataset)
return
model_without_ddp = model
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment