b87368e1e7fd832b505db9cc08015ac7af8f95de,VAE/main.py,,test,#Any#,115

Before Change

        test_loss += loss_function(recon_batch_data, batch_data, mu, logvar)

    test_loss = test_loss.data[0] / test_data.size(0)
    print("TEST SET RESULTS:" + " " * 20)
    print("Average loss: {:.4f}".format(test_loss))


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

After Change

def test(epoch):
    model.eval()
    test_loss = 0
    for batch in test_loader:
        batch = Variable(batch)

Instances