diff --git a/examples/benchmarks.py b/examples/benchmarks.py index bf204b4865..b5eec4566a 100644 --- a/examples/benchmarks.py +++ b/examples/benchmarks.py @@ -532,7 +532,6 @@ def _compute_tensorflow( except tf.errors.ResourceExhaustedError as e: print("Doesn't fit on GPU.", e) - torch.cuda.empty_cache() dictionary[model_name]["results"][batch_size][slice_size] = "N/A" dictionary[model_name]["memory"][batch_size][slice_size] = "N/A" return dictionary