From 46300822c8d070c0fb543971c7515eb9d2aa4ff5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kai=20Li=20=28=E6=9D=8E=E5=87=AF=29?= Date: Mon, 23 Sep 2024 21:40:15 +0800 Subject: [PATCH] Update audio_litmodule.py --- look2hear/system/audio_litmodule.py | 1 + 1 file changed, 1 insertion(+) 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