type=int,
help="Decide noise size!!!")
parser.add_argument("-b",
"--batch_size",
type=int,
help="Decide batch_size!!!")
parser.add_argument("-s",
"--save_period",
type=int,
help="Decide whether to store weights every few cycles")
parser.add_argument("-d",
"--dataset",
type=str,
help="select the dataset : MNIST? CIFAR10? ImageNet?")
args = parser.parse_args()
if args.state:
print args.state
Network.DCGAN(epoch=args.epoch,
noise_size=args.noise_size,
batch_size=args.batch_size,
save_period=args.save_period,
dataset=args.dataset)
else:
#dataset : MNIST, CIFAR10 , Imagenet
Network.DCGAN(epoch=1,
noise_size=100,
batch_size=128,
save_period=100,
dataset='CIFAR10')
