format

mindee · Jan 22, 2025 · 87c25e1 · 87c25e1
1 parent e9fc71d
commit 87c25e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/references/classification/train_pytorch_character.py b/references/classification/train_pytorch_character.py
@@ -145,6 +145,7 @@ def fit_one_epoch(model, train_loader, batch_transforms, optimizer, scheduler, a
 
         pbar.set_description(f"Training loss: {train_loss.item():.6} | LR: {last_lr:.6}")
         log(train_loss=train_loss.item(), lr=last_lr)
+
         epoch_train_loss += train_loss.item()
         batch_cnt += 1