fix

1 files changed, 1 insertions, 1 deletions

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index 9a8620f..2e72dee 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -352,7 +352,7 @@ def train(
                     ({100.0 * batch_idx / len(train_loader)}%)]\t \
                     Loss: {loss.item()}"
             )
-            return loss.item()
+        return loss.item()
 
 def test(model, device, test_loader, criterion):
     """Test"""