b87368e1e7fd832b505db9cc08015ac7af8f95de,VAE/main.py,,train,#Any#,94

Before Change

        loss = loss.data[0]
        optimizer.step()
        if i % 10 == 0:
            print("Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.4f}".format(
                epoch,
                i + BATCH_SIZE, training_data.size(0),
                float(i + BATCH_SIZE) / training_data.size(0) * 100,
                loss / BATCH_SIZE))


def test(epoch):

After Change

    model.train()
    train_loss = 0
    for batch in train_loader:
        batch = Variable(batch)

        optimizer.zero_grad()
        recon_batch, mu, logvar = model(batch)
        loss = loss_function(recon_batch, batch, mu, logvar)
        loss.backward()
        train_loss += loss
        optimizer.step()

    print("====> Epoch: {} Loss: {:.4f}".format(
          epoch,

Instances