cce8409beb4c1b662de2b9635cd723993184be23,torch_geometric/datasets/utils/faust_monet.py,,,#,28

Before Change

index = torch.stack([row, col], dim=0)
value = torch.stack([rho, theta], dim=1)
adj = torch.sparse.FloatTensor(index, value, torch.Size([6890, 6890, 2]))

outputs = []
for i in range(0, 100):
    file_path = os.path.join(geo_path, "tr_reg_{:03d}.mat".format(i))

After Change

values = []
slices = [0]
for i in range(80, 100):
    print(i)
    file_path = os.path.join(patch_path, "tr_reg_{:03d}.mat".format(i))
    out = read_matlab_sparse(file_path, "patch_coord")
    rho = out[:, :out.shape[0]].tocoo()
    theta = out[:, out.shape[0]:].tocoo()

Instances