cbcc3f315c653706ceded6ba42f222616322a3f2,ch03/03_atari_gan.py,,,#,132

Before Change

    true_labels_v = Variable(torch.FloatTensor([1.0] * BATCH_SIZE))
    fake_labels_v = Variable(torch.FloatTensor([0.0] * BATCH_SIZE))
    if args.cuda:
        true_labels_v = true_labels_v.cuda()
        fake_labels_v = fake_labels_v.cuda()

    for batch_v in iterate_batches(envs):
        // generate extra fake samples, input is 4D: batch, filters, x, y

After Change

    dis_losses = []
    iter_no = 0

    true_labels_v = torch.ones(BATCH_SIZE, dtype=torch.float32, device=device)
    fake_labels_v = torch.zeros(BATCH_SIZE, dtype=torch.float32, device=device)

    for batch_v in iterate_batches(envs):
        // generate extra fake samples, input is 4D: batch, filters, x, y

Instances