train.prototxt for flying chairs
kindloaf opened this issue · comments
Hong Chen commented
I'm trying to train flownet_S using flying chairs data.
The doc says train.prototxt should be created based on train.prototxt.template, and "make sure the correct parts of the network are enabled by setting/adding loss weights and blob learning rates."
I did the following changes, do they look good / complete?
$ diff FlowNet2-S_train.prototxt.template train.prototxt
14c14
< source: "../../data/FlyingThings3D_release_TRAIN_lmdb"
---
> source: "../../data/FlyingChairs_release_lmdb"
43c43
< source: "../../data/FlyingThings3D_release_TEST_lmdb"
---
> source: "../../data/FlyingChairs_release_lmdb"
214,215c214,215
< crop_width: 768
< crop_height: 384
---
> crop_width: 448
> crop_height: 320
310,311c310,311
< crop_width: 768
< crop_height: 384
---
> crop_width: 448
> crop_height: 320
331,332c331,332
< crop_width: 768
< crop_height: 384
---
> crop_width: 448
> crop_height: 320
Nikolaus Mayer commented
I think that should work for an "S" net on FlyingChairs, yes.
Hong Chen commented
Cool. Thanks.