3e11eb52b9b370bf9df6bf396e91ef94bf4055b8,examples/mnist_superpixel_75.py,,,#,6

Before Change

print(input.size())
print(len(adj))
print(len(position))
print(adj[].size())

After Change

    path, train=False, transform=Compose([Graclus(2), PolarAdj()]))

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

for batch, ((input, adj, position), target) in enumerate(test_loader):
    print(batch)
    // print(len(target))
    // print(input.size())

Instances