ACGAN-PyTorch icon indicating copy to clipboard operation
ACGAN-PyTorch copied to clipboard

ImageNet training parameters

Open Tushar-N opened this issue 7 years ago • 0 comments

Would it be possible to provide a run script for training on ImageNet, similar to the one you provided for CIFAR:

python main.py --outf=/your/output/file/name --niter=500 --batchSize=100 --cuda --dataset=cifar10 --imageSize=32 --dataroot=/data/path/to/cifar10 --gpu=0

I assume this should be okay, but I didn't want to guess any of the params (lr, niter, ngf, ndf etc.) either.

python main.py --outf=/your/output/file/name --niter=500 --batchSize=100 --cuda --dataset=imagenet --imageSize=128 --dataroot=/data/path/to/imagenet --num_classes=1000 --nz=1100 --gpu=0

What worked best in your experiments?

Tushar-N avatar Feb 20 '18 23:02 Tushar-N