add dummy jupyter notebook flag

Author: ellisbrown

.gitignore

@@ -123,3 +123,6 @@ test_data_aug.py
 # temp checkout soln
 data/datasets/
 data/ssd_dataloader.py
+
+# pylint
+.pylintrc

eval.py

@@ -45,7 +45,7 @@ def str2bool(v):
 parser.add_argument('--cuda', default=True, type=str2bool,
                     help='Use cuda to train model')
 parser.add_argument('--voc_root', default=VOCroot, help='Location of VOC root directory')
-
+parser.add_argument('-f', default=None, type=str, help="Dummy arg so we can load in Jupyter Notebooks")
 args = parser.parse_args()
 
 if not os.path.exists(args.save_folder):

test.py

@@ -23,6 +23,7 @@
 parser.add_argument('--cuda', default=True, type=bool,
                     help='Use cuda to train model')
 parser.add_argument('--voc_root', default=VOC_ROOT, help='Location of VOC root directory')
+parser.add_argument('-f', default=None, type=str, help="Dummy arg so we can load in Jupyter Notebooks")
 args = parser.parse_args()
 
 if args.cuda and torch.cuda.is_available():

train.py

@@ -38,6 +38,7 @@ def str2bool(v):
 parser.add_argument('--send_images_to_visdom', type=str2bool, default=True, help='Sample a random image from every 10th batch, send it to visdom after augmentations step')
 parser.add_argument('--save_folder', default='weights/', help='Directory for saving checkpoint models')
 parser.add_argument('--dataset_root', default=COCO_ROOT, help='Dataset root directory path')
+parser.add_argument('-f', default=None, type=str, help="Dummy arg so we can load in Jupyter Notebooks")
 args = parser.parse_args()
 
 if args.cuda and torch.cuda.is_available():