def main():
args = get_args()
envs, args = build_envs(args)
transfer_GAN = GAN(args.num_stack)
if args.cuda:
transfer_GAN.cuda()
print(transfer_GAN)
test_GAN(transfer_GAN, envs, args)
def main():
args = get_args()
replay_buffer = ReplayMemory(args.replay_size)
envs, args = build_envs(args)
transfer_GAN = GAN(args.num_stack, args)
if args.cuda:
transfer_GAN.cuda()
#print(transfer_GAN)
train_GAN(transfer_GAN, envs, replay_buffer, args)
