From 5202ddfbe9936b1245559954a96525db59b1bc3d Mon Sep 17 00:00:00 2001 From: "Cpt.Captain" Date: Tue, 22 Feb 2022 13:28:09 +0100 Subject: [PATCH] Add options for batch size, double heads and irl data --- co/args.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/co/args.py b/co/args.py index 19fe75e..8667dd4 100644 --- a/co/args.py +++ b/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()