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)