Update args to validation_seed and validation_split

This commit is contained in:
rockerBOO
2023-11-05 02:07:39 -05:00
parent 3de9e6c443
commit a93c524b3a

View File

@@ -1099,12 +1099,17 @@ def setup_parser() -> argparse.ArgumentParser:
action="store_true",
help="do not use fp16/bf16 VAE in mixed precision (use float VAE) / mixed precisionでも fp16/bf16 VAEを使わずfloat VAEを使う",
)
parser.add_argument(
"--validation_ratio",
"--validation_seed",
type=int,
default=None,
help="Validation seed"
)
parser.add_argument(
"--validation_split",
type=float,
default=0.0,
help="Ratio for validation images out of the training dataset"
help="Split for validation images out of the training dataset"
)
return parser