Skip to content
Permalink
Browse files

transform on ref

  • Loading branch information...
rusty1s committed May 15, 2019
1 parent c12c32a commit c0c4ea4b2d77ba96b50346d907132d0029150821
Showing with 8 additions and 1 deletion.
  1. +8 −1 torch_geometric/datasets/shrec2016.py
@@ -72,10 +72,17 @@ def __init__(self,
self.cat = category.lower()
super(SHREC2016, self).__init__(root, transform, pre_transform,
pre_filter)
self.ref = torch.load(self.processed_paths[0])
self.__ref__ = torch.load(self.processed_paths[0])
path = self.processed_paths[1] if train else self.processed_paths[2]
self.data, self.slices = torch.load(path)

@property
def ref(self):
ref = self.__ref__
if self.transform is not None:
ref = self.transform(ref)
return ref

@property
def raw_file_names(self):
return ['training', 'test']

0 comments on commit c0c4ea4

Please sign in to comment.
You can’t perform that action at this time.