Initial release

Jonathunky · Oct 29, 2023 · d966a05 · d966a05
1 parent f54f9e1
commit d966a05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/u2net_train.py b/u2net_train.py
@@ -147,9 +147,9 @@ def train_model(net, optimizer, dataloader, device, epoch_num, save_frq):
 
             if iteration_count % len(dataloader) == 0:
                 if epoch + 1 < epoch_num:
-                    print("Checkpoint saved. Loading next crops…")
+                    print("Checkpoint saved. Loading next crops…\n")
                 else:
-                    print("Final checkpoint")
+                    print("Final checkpoint\n")
 
         save_checkpoint(
             {