32c7386aef93737926069ee284d827f8e954e086,vae/main.py,,test,#Any#,122

Before Change

        recon_batch, mu, logvar = model(batch)
        test_loss += loss_function(recon_batch, batch, mu, logvar)

    test_loss = test_loss.data[0] / test_data.size(0)
    print("====> Test set results: {:.4f}".format(test_loss))


for epoch in range(1, NUM_EPOCHS + 1):

After Change

    for data, _ in test_loader:
        data = Variable(data, volatile=True)
        recon_batch, mu, logvar = model(data)
        test_loss += loss_function(recon_batch, data, mu, logvar).data[0]

    test_loss /= len(test_loader.dataset)
    print("====> Test set loss: {:.4f}".format(test_loss))

Instances