richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 921 forks source link

Training to long #68

Open bastienVR2 opened 5 years ago

bastienVR2 commented 5 years ago

Hello,

I have the following situation : I launched the training on my own data folder but for the moment, it takes 64 hours !!

And others things : I have this message "Train net output #0: loss8_313=4914.71 (* 1=4978.29 I0414 17:17:49.225188 14357 sgd_solver.cpp:112] Iteration 1331; lr=3.16e-5"

This lr is always the same for any iteration.

Is that normal that the training take long time in a data folder of only 9 images ??

PS : I'm using CPU-configuration for caffe.