Remove gc.collect() upon every model compute run. (pytorch#140)

eellison · Apr 14, 2022 · 4fd850c · 4fd850c
1 parent a460bde
commit 4fd850c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torchbench.py b/torchbench.py
@@ -162,7 +162,6 @@ def load_model(device, model_name, is_training, use_eval_mode):
 
 def timed(model, model_iter_fn, example_inputs, times=1, return_result=False):
     synchronize()
-    gc.collect()
     torch.manual_seed(1337)
     t0 = time.perf_counter()
     # Dont collect outputs to correctly measure timing