Skip to content

Commit 772df62

Browse files
committed
update README
1 parent 63909b5 commit 772df62

2 files changed

Lines changed: 7 additions & 10 deletions

File tree

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,11 @@ To train, use the following command format:
2121
python ./tools/train.py
2222
--model CSFlow
2323
--dataset Chairs
24+
--data_root $YOUR_DATA_PATH$
2425
--batch_size 1
2526
--name csflow-test
2627
--validation Sintel
27-
--gpus 0
28+
--val_Sintel_root $YOUR_DATA_PATH$
2829
--num_steps 100
2930
--lr 0.0004
3031
--image_size 368 496

tools/train.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)