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

Resume Training #92

Open amoazeni75 opened 3 years ago

amoazeni75 commented 3 years ago

Hello,

I wanted to resume training my model. As the README file stated, the command to resume training is: ./train/train_resume.sh ./train/models/colornet_iter_[ITERNUMBER].solverstate [GPU_ID] However, if you look at the train_resume.sh file, you see that the order of input argument is not consistent with the example provided. You should use this command: ./train/train_resume.sh [GPU_ID] ./train/models/colornet_iter_[ITERNUMBER].solverstate