search

acbull / pyHGT

Code for "Heterogeneous Graph Transformer" (WWW'20), which is based on pytorch_geometric
MIT License
775 stars 162 forks source link

MemoryError #24

Closed empty-id closed 3 years ago

empty-id commented 3 years ago

Hi, sorry to bother you, but I am facing a very weird problem.

from pyHGT.data import *
graph = renamed_load(open('preprocessed_OAG/graph_CS.pk', 'rb'))
print(type(graph))

When I run the above three lines of code. It runs normally. However, when I run the code train_paper_venue.py, I get a MemoryError at the renamed_load line.

Traceback (most recent call last):
  File "train_paper_venue.py", line 75, in <module>
    graph = renamed_load(open('preprocessed_OAG/graph_CS.pk', 'rb'))
  File "/home/usr/pyHGT-master/OAG/pyHGT/data.py", line 269, in renamed_load
    return RenameUnpickler(file_obj).load()
  File "/home/usr/anaconda3/envs/pyhgt/lib/python3.7/site-packages/dill/_dill.py", line 481, in load
    obj = StockUnpickler.load(self)
MemoryError

Could you help me?

update:

When I run these four lines of code, I also meets MemoryError:

from pyHGT.data import *
from pyHGT.model import *
graph = renamed_load(open('preprocessed_OAG/graph_CS.pk', 'rb'))
print(type(graph))

So what's wrong with pyHGT.model?

final update:

After downgrading pyG from 1.6.1 to 1.3.2, it's ok now...