Cpt.Captain/connecting_the_dots
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add options for batch size, double heads and irl data

Browse Source
This commit is contained in:
Cpt.Captain 2022-02-22 13:28:09 +01:00
parent db0a8523e5
commit 5202ddfbe9
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
co/args.py
View File

@ -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()