We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
LinkNeighborLoader
multiprocessing.spawn
1 parent b09990e commit c9608f1Copy full SHA for c9608f1
torch_geometric/loader/link_loader.py
@@ -100,8 +100,12 @@ def __init__(
100
self.data = data
101
self.edge_type = edge_type
102
self.link_sampler = link_sampler
103
- self.input_data = InputData(edge_label_index[0], edge_label_index[1],
104
- edge_label, edge_label_time)
+ self.input_data = InputData(
+ edge_label_index[0].clone(),
105
+ edge_label_index[1].clone(),
106
+ edge_label,
107
+ edge_label_time,
108
+ )
109
self.neg_sampling_ratio = neg_sampling_ratio
110
self.transform = transform
111
self.filter_per_worker = filter_per_worker
0 commit comments