finetuning - Githubissues

minivision-ai / photo2cartoon

人像卡通化探索项目 (photo-to-cartoon translation project)

MIT License

3.94k stars 763 forks source link

finetuning #40

Closed ak9250 closed 3 years ago

ak9250 commented 4 years ago

how can I finetune the existing photo2cartoon weights, I loaded the weights but then the training starts from 0 iterations

hao-qiang commented 4 years ago

You can finetune with '--pretrained_weights', and step will start from 0.

ak9250 commented 4 years ago

@hao-qiang ok that is kind of confusing compared to the original ugatit when you load the weights training starts from the last epoch of the saved model, can there be some indication that it is starting from the previous checkpoint?

hao-qiang commented 4 years ago

Please refer to： https://github.com/minivision-ai/photo2cartoon/blob/7590ed0396e648fb7437cd98a5cc59289d4ba5c1/models/UGATIT_sadalin_hourglass.py#L137-L146 You can change pretrained weights' name and get step from it.

ak9250 commented 4 years ago

@hao-qiang I tried that but training start at 0