Commit 67267821 authored by Mircea Cimpoi's avatar Mircea Cimpoi Committed by Facebook GitHub Bot
Browse files

Run zoomer profiling

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/518

Enable profiling for eval step only, not on every eval (which can be called during training)

Reviewed By: frabu6

Differential Revision: D44535915

fbshipit-source-id: 4497a3f74f5d751277df9ed41bc9bf21056341c4
parent c4b2d09c
......@@ -385,7 +385,8 @@ class Detectron2GoRunner(D2GoDataAPIMixIn, BaseRunner):
)
attach_profilers(cfg, model)
prepare_fb_model_for_eval(cfg, model)
if is_final:
prepare_fb_model_for_eval(cfg, model)
results = OrderedDict()
results[model_tag] = OrderedDict()
......
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