@@ -57,8 +57,7 @@ def parse_args():
5757 default = [384 , 512 ])
5858 parser .add_argument (
5959 '--data_root' ,
60- help = 'Root of the current training datasets' ,
61- default = 'E://Labs//DataSet//FlyingChairs_release//data' )
60+ help = 'Root of the current training datasets' )
6261 parser .add_argument (
6362 '--validation' ,
6463 type = str ,
@@ -67,16 +66,13 @@ def parse_args():
6766 choices = ['KITTI' , 'Sintel' , 'Chairs' ])
6867 parser .add_argument (
6968 '--val_Chairs_root' ,
70- help = 'Root of the Chairs validation datasets' ,
71- default = 'E://Labs//DataSet//FlyingChairs_release//data' )
69+ help = 'Root of the Chairs validation datasets' )
7270 parser .add_argument (
7371 '--val_Sintel_root' ,
74- help = 'Root of the Sintel validation datasets' ,
75- default = 'D://DataSet//MPI-Sintel-complete' )
72+ help = 'Root of the Sintel validation datasets' )
7673 parser .add_argument (
7774 '--val_KITTI_root' ,
78- help = 'Root of the KITTI validation datasets' ,
79- default = None )
75+ help = 'Root of the KITTI validation datasets' )
8076 parser .add_argument ('--DEVICE' , help = 'The using device' , default = 'cuda' )
8177 parser .add_argument (
8278 '--lr' , type = float , help = 'Learning rate' , default = 0.00002 )
@@ -94,7 +90,7 @@ def parse_args():
9490 '--change_gpu' ,
9591 help = 'train on cuda device but not cuda:0' ,
9692 action = 'store_true' )
97- parser .add_argument ('--gpus' , type = int , nargs = '+' , default = [ 0 , 1 ] )
93+ parser .add_argument ('--gpus' , type = int , nargs = '+' , default = 0 )
9894 parser .add_argument (
9995 '--iters' ,
10096 type = int ,
0 commit comments