Unverified Commit 8becb732 authored by Nitish Shirish Keskar's avatar Nitish Shirish Keskar Committed by GitHub
Browse files

removing torch.cuda.empty_cache() from TF function (#3267)

torch.cuda.empty_cache() was being called from a TF function (even when torch is unavailable)
not sure any replacement is needed if TF OOMs
parent ecfd3363
...@@ -532,7 +532,6 @@ def _compute_tensorflow( ...@@ -532,7 +532,6 @@ def _compute_tensorflow(
except tf.errors.ResourceExhaustedError as e: except tf.errors.ResourceExhaustedError as e:
print("Doesn't fit on GPU.", 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]["results"][batch_size][slice_size] = "N/A"
dictionary[model_name]["memory"][batch_size][slice_size] = "N/A" dictionary[model_name]["memory"][batch_size][slice_size] = "N/A"
return dictionary return dictionary
......
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