ACGAN-PyTorch
ACGAN-PyTorch copied to clipboard
ImageNet training parameters
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?