Model training Loss graph about colorization-pytorch HOT 6 OPEN

The problem was with the initialization. In the original paper, I used k-means initialization, but didn't have the code for pytorch.

I just pushed a change. If you rerun python make_ilsvrc_dataset.py, it will make a directory dataset/ilsvrc2012/train_small/, which contains 10 imagenet classes. Then rerun bash scripts/train_siggraph.sh again. It will first run 100 epochs on this small training set as "initialization", before training on the full training set.

It does still train quite slowly. Better networks/loss functions would be a good update to the project.

Thanks for your patience.

from colorization-pytorch.

Thanks for the message. I'll look into it in the next week or so.

from colorization-pytorch.

My apologies for the delay. I'm caught up with a few things and will have to look into this more after CVPR deadline (mid-Nov).

from colorization-pytorch.

Thanks.

from colorization-pytorch.

Hello author,
I have the same problem above, what is the solution?
Thanks in advance.

from colorization-pytorch.

I believe I did not do the initialization properly. I am looking into it. Thanks for your patience!

from colorization-pytorch.