search

jantic / DeOldify

A Deep Learning based project for colorizing and restoring old images (and video!)
MIT License
17.95k stars 2.55k forks source link

How to train a model? #242

Closed keto33 closed 4 years ago

keto33 commented 4 years ago

How can I train a model or continue training the available model?

jantic commented 4 years ago

If you have the prerequisite background in deep learning (fastai would be ideal), then what's in the readme along with the Jupyter notebooks should suffice. That being said, training NoGAN is particularly challenging even with a solid background (it's tricky to know when to stop, namely).

If you don't have that background I can't really help you and you won't get too far until something goes wrong. It's not so much of a "follow this recipe" sort of thing unfortunately.