no eval detection flag

Author: ylabbe

cosypose/scripts/run_detector_training.py

@@ -13,6 +13,7 @@
     parser.add_argument('--config', default='', type=str)
     parser.add_argument('--debug', action='store_true')
     parser.add_argument('--resume', default='', type=str)
+    parser.add_argument('--no-eval', action='store_true')
     args = parser.parse_args()
 
     cfg = argparse.ArgumentParser('').parse_args([])
@@ -104,6 +105,9 @@
         raise ValueError(args.config)
     cfg.val_ds_names = cfg.train_ds_names
 
+    if args.no_eval:
+        cfg.test_ds_names = []
+
     cfg.run_id = f'detector-{args.config}-{run_comment}-{N_RAND}'
 
     if args.debug: