How many epochs to train for CIFAR-10

openai / pixel-cnn

Code for the paper "PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications"

https://arxiv.org/abs/1701.05517

Other

1.92k stars 436 forks source link

How many epochs to train for CIFAR-10 #57

Open erikdao opened 2 years ago

erikdao commented 2 years ago

Hi, I'm wondering how many epochs should one train PixcelCNN++ from scratch on CIFAR-10 o get a reasonable result (e.g., Inception score around ~5.5)? I can see in the train.py, the max_epochs is defaulted to 5000 epochs. Does one actually need to train that much?