diff options
Diffstat (limited to 'swr2_asr/train.py')
| -rw-r--r-- | swr2_asr/train.py | 2 |
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""" |