Opt fails to be downloaded

[yaxingwang]

Hi

CUDA_VISIBLE_DEVICES=O python src/main.py -t -hdf5 -l -batch_stat -metrics is fid prdc -ref "test" -best --project Cifar10-wgangp -cfg src/configs/CIFAR10/WGAN-GP.yaml -ckpt author_trained_models/cifar10/wgangp/WGAN-GP-train-2020_09_16_14_17_00 -data cifar10 -save results

I suffer from issue: RuntimeError: The size of tensor a (3) must match the size of tensor b (128) at non-singleton dimension 3, which is due to the case the size of 'exp_avg' are different to the ones when building model.