pix2pix icon indicating copy to clipboard operation
pix2pix copied to clipboard

How to train on a pretrained model?

Open somiltg opened this issue 4 years ago • 3 comments

We are trying to transfer learn on pix2pix model by using the edges2shoes dataset and using --continue_train to train on our dataset. However, we receive an error regarding Discriminator saying that checkpoint is not present. Why is the checkpoint for discriminator not present in the pretrained model? Is it because it should only be used for testing and not training?

somiltg avatar Dec 01 '19 21:12 somiltg

We didn't save the discriminator. You need to train the discriminator from scratch.

junyanz avatar Dec 02 '19 20:12 junyanz

Do you mean we need to rerun the network on the entire Zappos dataset, because until we provide the checkpoint, we cannot use the pretrained model further. Can you suggest how we can go foorward with the method?

somiltg avatar Dec 02 '19 22:12 somiltg

We did provide the generator's weights. To fine-tune the model on your own dataset, you can initialize the generator with our weights, and train a discriminator from scratch on your own dataset. You don't need the original dataset.

junyanz avatar Dec 03 '19 05:12 junyanz