Update submission.py

JiaRenChang · web-flow · commit 43e8eb2d4735 · 2018-03-23T19:22:33.000+08:00
diff --git a/submission.py b/submission.py
@@ -43,7 +43,7 @@
 if args.cuda:
     torch.cuda.manual_seed(args.seed)
 
-if args.KITTI == 2015:
+if args.KITTI == '2015':
    from dataloader import KITTI_submission_loader as DA
 else:
    from dataloader import KITTI_submission_loader2012 as DA  
