/GAN_beginner

GAN pytorch for beginner

Primary LanguagePython

Deep Learning Beginner

[CNN pytorch for Beginner]

-train script-

python main.py --model_type CNN --dataset mnist (or cifar10) --num_workers 4 --comment train_1t

[RNN pytorch for Beginner]

[GAN pytorch for beginner]

-GAN-

python main.py --model_type GAN --dataset mnist (or cifar10) --num_workers 4 --comment GAN_1t

-WGAN-

python main.py --model_type WGAN --dataset mnist (or cifar10) --num_workers 4 --comment GAN_1t

-DiracGAN-

python main.py --model_type DiracGAN --dataset mnist (or cifar10) --num_workers 4 --comment GAN_1t