Browse Source

whitespace

master
Florin Tobler 5 months ago
parent
commit
3218e7eb63
  1. 2
      inference_profile_test.py

2
inference_profile_test.py

@ -10,7 +10,7 @@ def empty_cuda():
while True:
gc.collect()
torch.cuda.empty_cache()
time.sleep(0.5)
time.sleep(0.5)
vram = nvidia_smi.get_gpu_stats()["memory_used"]
print("vram: %d MB" % vram)
if vram < 200:

Loading…
Cancel
Save