Training code for ImageNet.

The purpose of this code is to pre-training feature extractors, discriminators, perceptual loss, etc.

Use train.py.

Example command

python train.py imagenet --arch torchvision.vgg11_bn --data-dir /data/dataset/ImageNet --model-dir ./models/vgg11 --norm center --num-samples 50000 --batch-size 64

You should prepare ImageNet dataset yourself.