Upload 2 files

mod.py

@@ -78,7 +78,7 @@ def clear_cache():
     try:
         torch.cuda.empty_cache()
         #torch.cuda.reset_max_memory_allocated()
-        torch.cuda.reset_peak_memory_stats()
+        #torch.cuda.reset_peak_memory_stats()
         gc.collect()
     except Exception as e:
         print(e)