Merge pull request #480 from yanhuifair/main

fix print "saving" and "epoch" in newline
2026-04-08 22:35:09 +00:00 · 2023-05-11 21:05:37 +09:00
parent 3b1af3f1a6 b08154dc36
commit 968bbd2f47
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最初の状態