diff --git a/look2hear/system/audio_litmodule.py b/look2hear/system/audio_litmodule.py
index 8faa6df..5888485 100644
--- a/look2hear/system/audio_litmodule.py
+++ b/look2hear/system/audio_litmodule.py
@@ -158,6 +158,7 @@ def on_validation_epoch_end(self):
         )
 
         self.validation_step_outputs.clear()  # free memory
+        torch.cuda.empty_cache()
         
     def test_step(self, batch, batch_nb):
         mixtures, targets = batch