Fix alexnet setup only (LLNL#2124)

* modify LC paths away from brainusr directory * modify LC paths away from brainusr directory * bugfix of alexnet.py setup-only flag Co-authored-by: bburnett12 <[email protected]>
graham63 · Jun 28, 2022 · 6fc27da · 6fc27da
1 parent 13d48be
commit 6fc27da
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/applications/vision/alexnet.py b/applications/vision/alexnet.py
@@ -23,9 +23,6 @@
     '--num-classes', action='store', default=1000, type=int,
     help='number of ImageNet classes (default: 1000)', metavar='NUM')
 lbann.contrib.args.add_optimizer_arguments(parser)
-parser.add_argument(
-    '--setup_only', action='store_true',
-    help='setup LBANN experiment without running it')
 args = parser.parse_args()
 
 # Due to a data reader limitation, the actual model realization must be
@@ -76,5 +73,4 @@
 kwargs = lbann.contrib.args.get_scheduler_kwargs(args)
 lbann.contrib.launcher.run(trainer, model, data_reader, opt,
                            job_name=args.job_name,
-                           setup_only=args.setup_only,
                            **kwargs)