Unverified Commit cf01fe0b authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Fix name of new trainer option

parent 8451dc68
......@@ -223,7 +223,7 @@ def main(args):
seed_everything(args.seed)
config = model_config(
args.model_preset,
args.config_preset,
train=True,
low_prec=(args.precision == "16")
)
......@@ -473,8 +473,8 @@ if __name__ == "__main__":
"--log_lr", action="store_true", default=False,
)
parser.add_argument(
"--model_preset", type=str, default="initial_training",
help='Model config setting. Choose e.g. "initial_training", "finetuning", "model_1", etc.'
"--config_preset", type=str, default="initial_training",
help='Config setting. Choose e.g. "initial_training", "finetuning", "model_1", etc.'
)
parser = pl.Trainer.add_argparse_args(parser)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment