fix print "saving" and "epoch" in newline

This commit is contained in:
Fair
2023-05-07 02:51:01 +08:00
parent e6ad3cbc66
commit b08154dc36
11 changed files with 21 additions and 21 deletions

View File

@@ -266,7 +266,7 @@ def train(args):
accelerator.init_trackers("finetuning" if args.log_tracker_name is None else args.log_tracker_name)
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
for m in training_models: