Skip to content

Commit a8687f5

Browse files
committed
Update args to validation_seed and validation_split
1 parent e2b819f commit a8687f5

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

train_network.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1105,12 +1105,17 @@ def setup_parser() -> argparse.ArgumentParser:
11051105
action="store_true",
11061106
help="do not use fp16/bf16 VAE in mixed precision (use float VAE) / mixed precisionでも fp16/bf16 VAEを使わずfloat VAEを使う",
11071107
)
1108-
11091108
parser.add_argument(
1110-
"--validation_ratio",
1109+
"--validation_seed",
1110+
type=int,
1111+
default=None,
1112+
help="Validation seed"
1113+
)
1114+
parser.add_argument(
1115+
"--validation_split",
11111116
type=float,
11121117
default=0.0,
1113-
help="Ratio for validation images out of the training dataset"
1118+
help="Split for validation images out of the training dataset"
11141119
)
11151120

11161121
return parser

0 commit comments

Comments
 (0)