update

Author: bhimrazy

tests/tests_pytorch/trainer/test_trainer.py

@@ -1750,6 +1750,7 @@ def on_train_epoch_start(self, trainer, *_):
     def current_memory():
         # before measuring the memory force release any leftover allocations, including CUDA tensors
         gc.collect()
+        torch.cuda.empty_cache()
         return torch.cuda.memory_allocated(0)
 
     model = TestModel()