796a6e55afacbaa0ada91a29926c6635f9cf85c4,examples/cuneiform.py,,,#,109

Before Change

for epoch in range(1, 201):
    train(epoch)
    test(epoch, train_loader, "Train Accuracy")
    test(epoch, test_loader, " Test Accuracy")

After Change

    elif i == split.size(0) - 2:
        train_split = perm[:split[i]]
    else:
        train_split = torch.cat([perm[:split[i]], perm[split[i + 1]:]])

    train_dataset.split = train_split
    test_dataset.split = test_split

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

    model.conv1.reset_parameters()

Instances