Skip to content

Commit

Permalink
obsolete arg
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Aug 8, 2020
1 parent 934cea2 commit ef2839c
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions pl_examples/domain_templates/imagenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def main(args: Namespace) -> None:
args.workers = int(args.workers / max(1, args.gpus))

model = ImageNetLightningModel(**vars(args))
trainer = pl.Trainer.from_argparse_args(args, default_root_dir=args.save_path)
trainer = pl.Trainer.from_argparse_args(args)

if args.evaluate:
trainer.test(model)
Expand All @@ -251,8 +251,6 @@ def run_cli():
parent_parser = pl.Trainer.add_argparse_args(parent_parser)
parent_parser.add_argument('--data-path', metavar='DIR', type=str,
help='path to dataset')
parent_parser.add_argument('--save-path', metavar='DIR', default=".", type=str,
help='path to save output')
parent_parser.add_argument('-e', '--evaluate', dest='evaluate', action='store_true',
help='evaluate model on validation set')
parent_parser.add_argument('--seed', type=int, default=42,
Expand Down

0 comments on commit ef2839c

Please sign in to comment.