richzhang/colorization

Resume Training

amoazeni75 opened this issue · 0 comments

amoazeni75 commented

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