Add options for batch size, double heads and irl data

co/args.py

@@ -14,7 +14,7 @@ def parse_args():
                               train with \'phge\' for the second stage with geometric loss',
                         default='ph', choices=['ph', 'phge'], type=str)
     parser.add_argument('--data_type',
-                        default='syn', choices=['syn'], type=str)
+                        default='syn', choices=['syn', 'irl'], type=str)
     #
     parser.add_argument('--cmd',
                         help='Start training or test',
@@ -55,6 +55,14 @@ def parse_args():
     parser.add_argument('--blend_im',
                         help='Parameter for adding texture',
                         default=0.6, type=float)
+    # FIXME this is pretty confusing, redo at some point
+    parser.add_argument('--no_double_heads',
+                        help='Don\'t use second head trained with disparity GT',
+                        action='store_false')
+
+    parser.add_argument('--batch_size',
+                        help='Batch size for train and test',
+                        default=6, type=int)
 
     args = parser.parse_args()