From 93c50cc657e0b1c5b8ef0b2eb6fe3b2277149745 Mon Sep 17 00:00:00 2001 From: Francisc Bungiu Date: Wed, 28 Jun 2023 06:43:17 -0700 Subject: [PATCH] Remove profiling of evaluation Summary: Deprecate prepare_fb_model_for_eval(). Reviewed By: miqueljubert Differential Revision: D47085783 fbshipit-source-id: 2b02a14be5cf8f503fab467c76b88f572c82275d --- d2go/runner/default_runner.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/d2go/runner/default_runner.py b/d2go/runner/default_runner.py index f615fff8..995ad2a4 100644 --- a/d2go/runner/default_runner.py +++ b/d2go/runner/default_runner.py @@ -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 @@ def _get_inference_dir_name(base_dir, inference_type, dataset_name): ) attach_profilers(cfg, model) - if is_final: - prepare_fb_model_for_eval(cfg, model) results = OrderedDict() results[model_tag] = OrderedDict()