Could you provide the pretrained model of discriminator of 256_shortcut1_inject1_none_hq?

elvisyjlin / AttGAN-PyTorch

AttGAN PyTorch Arbitrary Facial Attribute Editing: Only Change What You Want

MIT License

248 stars 61 forks source link

Could you provide the pretrained model of discriminator of 256_shortcut1_inject1_none_hq? #26

Open liang23333 opened 3 years ago

liang23333 commented 3 years ago

There is no pretrained model of 256_shortcut1_inject1_none_hq.

elvisyjlin commented 3 years ago

Can you access this google drive file? The file is 154MB. After you download and extract the zip, you will find it under 256_shortcut1_inject1_none_hq/checkpoint/weights.199.pth.

MouhamedAhmed commented 3 years ago

There is no pretrained discriminator in this drive.

elvisyjlin commented 3 years ago

The model is saved every 1000 iterations, so there are a bunch of model in the end. To reduce the disk storage they take, I only save the weights of the generator. They take around 300GB for AttGAN generators on Celeb-A tho. You can translate images with the pretrained on the cloud. If you need the weights of discriminator, please train it from scratch by yourself. Remember to set the save_internal larger, e.g. 10000, to prevent from out of storage. I'm sorry for any inconvenience about this.