How to run the scripts: python resnet-opt --data_dir=/path/to/imagenet --batch_size=1024 --num_iter=300