disable weighted captions in TI/XTI training

This commit is contained in:
Kohya S
2023-04-08 21:45:57 +09:00
parent a876f2d3fb
commit 6a5f87d874
3 changed files with 10 additions and 9 deletions

View File

@@ -603,7 +603,7 @@ def setup_parser() -> argparse.ArgumentParser:
train_util.add_training_arguments(parser, True)
train_util.add_optimizer_arguments(parser)
config_util.add_config_arguments(parser)
custom_train_functions.add_custom_train_arguments(parser)
custom_train_functions.add_custom_train_arguments(parser, False)
parser.add_argument(
"--save_model_as",