search

facebookresearch / pytorch_GAN_zoo

A mix of GAN implementations including progressive growing
BSD 3-Clause "New" or "Revised" License
1.62k stars 271 forks source link

Pretrained model for CIFAR 10 dataset? #63

Closed tianweiy closed 4 years ago

tianweiy commented 5 years ago

Is there any pretrained model for cifar 10 dataset? I tried to train using the code here but the loss diverges in the end and I can only get 5.2 inception score.

Molugan commented 5 years ago

Hi, Sorry for the delay.

There isn't a pretained model for CIFAR yet. But I can have a look.

Molugan commented 5 years ago

I've found a bug explaining partly the low score on CIFAR so far, after https://github.com/facebookresearch/pytorch_GAN_zoo/pull/72 you should easily get a 6.1 score. I'm working on the parameters to go up to 8.

tianweiy commented 5 years ago

Thanks. No hurry, I am using gan in TensorFlow in the meantime.

heartInsert commented 5 years ago

Thanks. No hurry, I am using gan in TensorFlow in the meantime.

How much time do you cost in training in cifar10 with 5.2 inception score ?

tianweiy commented 5 years ago

For the 5.2, I am trying to say that the loss diverges in the end so that current implementation is far from optimal. I suspect you can get a little higher inception score by training less time. It took about 2~3 days using 2 GPUs if I remember correctly.

heartInsert commented 5 years ago

God, it 's too much time for me

yifanjiang19 commented 5 years ago

@Molugan Can the current version reproduce the results (fid=8.80) for cifar-10?

JanineCHEN commented 4 years ago

@Molugan Can the current version reproduce the results (fid=8.80) for cifar-10?

@Molugan Having same question here, any insight would be greatly appreciated!