search

chshin10 / epinet

EPINET: A Fully-Convolutional Neural Network using Epipolar Geometry for Depth from Light Field Images
MIT License
88 stars 39 forks source link

Hello, some question about training code #7

Closed SamuelSue94 closed 5 years ago

SamuelSue94 commented 5 years ago

I use imgaug to do data_augmentation, and train this model (input 9x9 images), and noticed that every epoch in model.fit() cost 100+ seconds and I run this on 2 TITAN V,and after 2600 epochs the results are bad, so I just want to know how you train this model, do you have a plan to release the full code? Since I was in this trouble for almost one month, more details would be appreciated.

chshin10 commented 5 years ago

I upload train code now! It took more time than I thought. Sorry for the late.

And I think 1-day is enough time for good result(my environment: i7-7700, one gtx1080ti, 32GB ram)