Add options for batch size, double heads and irl data

master
Cpt.Captain 3 years ago
parent db0a8523e5
commit 5202ddfbe9
  1. 10
      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()

Loading…
Cancel
Save