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

Before Change

    if cuda:
        batch_data_t = batch_data_t.cuda()
    batch_data = Variable(batch_data_t, volatile=True)
    for i in range(0, test_data.size(0), TEST_BATCH_SIZE):
        print("Testing model: {}/{}".format(i, test_data.size(0)), end="\r")
        batch_data.data[:] = test_data[i:i + TEST_BATCH_SIZE]
        recon_batch_data, mu, logvar = model(batch_data)
        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))

After Change

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

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

Instances