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", action="store_true",
help="do not use fp16/bf16 VAE in mixed precision (use float VAE) / mixed precisionでも fp16/bf16 VAEを使わずfloat VAEを使う", help="do not use fp16/bf16 VAE in mixed precision (use float VAE) / mixed precisionでも fp16/bf16 VAEを使わずfloat VAEを使う",
) )
parser.add_argument( parser.add_argument(
"--validation_ratio", "--validation_seed",
type=int,
default=None,
help="Validation seed"
)
parser.add_argument(
"--validation_split",
type=float, type=float,
default=0.0, 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 return parser