Closed playma closed 5 years ago
You have a few options. If you want to start completely from scratch, then you should train ColorizeTrainingStable.ipynb. This is stated at the top of ColorizeTrainingVideo.ipynb:
It's assumed that there's a pretrained generator from the ColorizeTrainingStable notebook available at the specified path.
But this isn't the only way to get a pretrained generator. You can also download its weights from the readme, in the section that looks like this:
Pretrain Only Generator Weights artistic stable video
Pretrain Only Critic Weights artistic stable video
Download the stable ones if you want to start the ColorizeTrainingVideo.ipynb exactly like you would training ColorizeTrainingStable.ipynb on your own first. Or you download the video weights to get straight to GAN training and bypass pre-training on noise augmentation.
I am newbie. please mention how to place data in which format or folder. Please can you explain training process in detail ... or add new readme file or manual. because i have got an error on this line learn_gen.fit_one_cycle(1, pct_start=pct_start, max_lr=slice(5e-8,5e-5))
StopIteration Traceback (most recent call last)
@hammadalibutt Don't take this the wrong way but I can tell you right now that this won't be the end of the list of questions you'll wind up running into if you're asking me this. I'm not saying this to be mean but to be practical: Basic skills (debugging, reading code) are in order first before you can tackle running this training. I just don't have the time to walk you through all this.
Yes, you are right, but sorry for interruption again. Please, can you help me how to place my data set in your algorithm? another question is video trainer use image datasets?
I've already said no dude. I'm not sure what you think changed between now and then....
Thank you for the great work!
I have some questions
I want to use
DeOldify
to train a video model like pix2pix. Which notebook should I use? I found that it did not convert color images to black and white image as training data inColorizeTrainingVideo.ipynb
, but inColorizeTrainingArtistic.ipynb
andColorizeTrainingStable.ipynb
Should I train
ColorizeTrainingArtistic.ipynb
orColorizeTrainingStable.ipynb
first, and then train theColorizeTrainingVideo.ipynb
?