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

@@ -94,7 +94,7 @@ def split(args):
filename, ext = os.path.splitext(args.save_to)
model_file_name = filename + f"-{new_rank:04d}{ext}"
print(f"saving model to: {model_file_name}")
print(f"\nsaving model to: {model_file_name}")
save_to_file(model_file_name, state_dict, new_metadata)