ImbGAN Pretrain python pretrain.py --dataset MNIST --dataroot ./dataset --path ./ckpts train python train.py --dataset MNIST --dataroot ./dataset --path ./ckpts or python train_sub.py --dataset MNIST --dataroot ./dataset --path ./ckpts