Change training schedule

Author: yura-hb

diploma_thesis/simulator/simulator.py

@@ -398,10 +398,10 @@ def __log_time__(self, environment: simpy.Environment, log_tick: float):
 
             gc.collect()
 
-            # try:
-            #     torch.mps.empty_cache()
-            # except:
-            #     pass
+            try:
+                torch.mps.empty_cache()
+            except:
+                pass
 
             torch.cuda.empty_cache()