960518e2734a42f7da19dc3d4d49501004c037e1,mnist/main.py,,,#,54

Before Change

TEST_BATCH_SIZE = 1000
NUM_EPOCHS = 2

optimizer = optim.SGD(model, lr=1e-2, momentum=0.9)

def train(epoch):
    batch_data = Variable(torch.cuda.FloatTensor(BATCH_SIZE, 1, 28, 28), requires_grad=False)
    batch_targets = Variable(torch.cuda.FloatTensor(BATCH_SIZE), requires_grad=False)

After Change

def print_header(msg):
    print("===>", msg)

if not os.path.exists("data/processed/training.pt"):
    import data

// Data
print_header("Loading data")
with open("data/processed/training.pt", "rb") as f:
    training_set = torch.load(f)
with open("data/processed/test.pt", "rb") as f:

Instances