Commit b1e24e81 authored by Francisc Bungiu's avatar Francisc Bungiu Committed by Facebook GitHub Bot
Browse files

Remove profiling of evaluation

Summary: Deprecate prepare_fb_model_for_eval().

Reviewed By: miqueljubert

Differential Revision: D47085783

fbshipit-source-id: 34b7e822e9baa1f9f77a11d3497df7fb0463c955
parent 2de6546e
......@@ -168,11 +168,6 @@ def prepare_fb_model(cfg: CfgNode, model: torch.nn.Module) -> torch.nn.Module:
return model
@fb_overwritable()
def prepare_fb_model_for_eval(cfg: CfgNode, model: torch.nn.Module) -> torch.nn.Module:
return model
class BaseRunner(object):
def __init__(self):
identifier = f"D2Go.Runner.{self.__class__.__name__}"
......@@ -424,8 +419,6 @@ class Detectron2GoRunner(D2GoDataAPIMixIn, BaseRunner):
)
attach_profilers(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