add ja comment, restore arg for backward compat

This commit is contained in:
Kohya S
2023-04-17 21:58:55 +09:00
parent 01ebfc41f3
commit 8f6fc8daa1
4 changed files with 16 additions and 10 deletions

View File

@@ -73,7 +73,7 @@ def setup_parser() -> argparse.ArgumentParser:
parser.add_argument("--float", action='store_true',
help='save as float (checkpoint only) / float(float32)形式で保存するcheckpointのみ対応')
parser.add_argument("--save_precision_as", type=str, default="no", choices=["fp16", "bf16", "float"],
help="save precision")
help="save precision, do not specify with --fp16/--bf16/--float / 保存する精度、--fp16/--bf16/--floatと併用しないでください")
parser.add_argument("--epoch", type=int, default=0, help='epoch to write to checkpoint / checkpointに記録するepoch数の値')
parser.add_argument("--global_step", type=int, default=0,
help='global_step to write to checkpoint / checkpointに記録するglobal_stepの値')