ecd54bcead758c0dff7b275202f5f1d71bf5f5f6,examples/cuneiform.py,,,#,114

Before Change

    train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
    test_loader = DataLoader(test_dataset, batch_size=32, shuffle=True)

    model.conv1.reset_parameters()
    model.conv2.reset_parameters()
    model.fc1.reset_parameters()

    for epoch in range(1, 301):

After Change

    test_loader = DataLoader(test_dataset, batch_size=32, shuffle=True)

    accs_single = []
    for _ in range(10):
        model.conv1.reset_parameters()
        model.conv2.reset_parameters()
        model.fc1.reset_parameters()
        for epoch in range(1, 301):
            train(epoch)
        acc = test(epoch, test_loader, " Test Accuracy")
        accs_single.append(acc)
    mean = torch.FloatTensor(accs_single).mean()
    print("Mean", mean)
    accs.append(mean)

Instances