24387ddc838a9eb4273c03bf19e3f35587e3f201,implementations/wgan/wgan.py,,,#,88

Before Change

for epoch in range(opt.n_epochs):

    // Batch iterator
    data_iter = iter(dataloader)

    for i in range(len(data_iter) // opt.n_critic):
        // Train discriminator for n_critic times
        for _ in range(opt.n_critic):

After Change

batches_done = 0
for epoch in range(opt.n_epochs):

    for i, (imgs, _) in enumerate(dataloader):

        // Configure input
        real_imgs = Variable(imgs.type(Tensor))

Instances