b87368e1e7fd832b505db9cc08015ac7af8f95de,VAE/main.py,,,#,87

Before Change

TEST_BATCH_SIZE = 1000
NUM_EPOCHS = 2

optimizer = optim.Adam(model.parameters(), lr=1e-3)


def train(epoch):
    batch_data_t = torch.FloatTensor(BATCH_SIZE, 784)

After Change

    print("====> Test set results: {:.4f}".format(test_loss))


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

Instances