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

    if i == 0:
        train_split = perm[split[i + 1]:]
    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

Instances