interactive-deep-colorization icon indicating copy to clipboard operation
interactive-deep-colorization copied to clipboard

Published 20 hours ago • verified publisher icon junyanz

Fine-tune model

Open zeynelsamak opened this issue 8 years ago • 9 comments
trafficstars

Hello, I would like to fine-tune the model with new data. It would be great if you could share the train.prototxt. Thanks

zeynelsamak avatar Oct 09 '17 10:10 zeynelsamak

We are looking at it. Will upload one soon.

junyanz avatar Nov 22 '17 06:11 junyanz

thank you @junyanz, I look forward it

zeynelsamak avatar Nov 22 '17 10:11 zeynelsamak

Any updates regarding this @junyanz ?

j-w-shin avatar Apr 19 '18 02:04 j-w-shin

My apologies for the delay. It is on the list of todo's but I have been caught up with many things. I will get to it as soon as possible.

richzhang avatar Apr 19 '18 02:04 richzhang

@junyanz @richzhang any updates regarding this?

sushant21 avatar Jun 11 '18 10:06 sushant21

I'm working on a pytorch version for public release. Thanks for your patience.

richzhang avatar Jun 11 '18 15:06 richzhang

Great to hear that! Can you give a time estimate for this release, just to know if I can hold off my work till then?

sushant21 avatar Jun 12 '18 08:06 sushant21

Sorry for the delay. I am retraining it in pytorch now. Hopefully release in a few weeks. Thanks for your patience.

richzhang avatar Aug 08 '18 18:08 richzhang

A PyTorch reimplementation is in this repository. We updated this repository as well to support PyTorch models in the backend. Apologies again for the delay.

richzhang avatar Sep 04 '18 04:09 richzhang