fix print "saving" and "epoch" in newline

2026-04-08 22:35:09 +00:00 · 2023-05-07 02:51:01 +08:00
parent e6ad3cbc66
commit b08154dc36
11 changed files with 21 additions and 21 deletions
--- a/train_db.py
+++ b/train_db.py
@@ -240,7 +240,7 @@ def train(args):
    loss_list = []
    loss_total = 0.0
    for epoch in range(num_train_epochs):
-        print(f"epoch {epoch+1}/{num_train_epochs}")
+        print(f"\nepoch {epoch+1}/{num_train_epochs}")
        current_epoch.value = epoch + 1

        # 指定したステップ数までText Encoderを学習する：epoch最初の状態