how to modify train.prototxt

NVlabs / PWC-Net

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)

Other

1.63k stars 357 forks source link

how to modify train.prototxt #87

Closed shimhyeonwoo closed 5 years ago

shimhyeonwoo commented 5 years ago

when I train pwcnet using FlyingChairs with train.prototxt, It doesn't converge. I modified train.prototxt :

flyingchairs lmdb and split.list directory
crop size : 320x448, 384x448 tried both crop size, both results were similar
batch size : 4 (used 2 gpu, then effective batch size is 8)

after train, I tested the model on FlyingChairs test set (labeled as 2 in test_train_split.list) EPE was 10.9164379646 anyone can help me?? any idea you give will be thankful!!

shimhyeonwoo commented 5 years ago

I guess it was initialization problem it converges well good luck!