search

PaddlePaddle / PGL

Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework based on PaddlePaddle
Apache License 2.0
1.57k stars 309 forks source link

erniesage example raise generator wrong #167

Closed selinplus closed 8 months ago

selinplus commented 3 years ago

Traceback (most recent call last): File "link_predict.py", line 274, in train(config) File "link_predict.py", line 169, in train train_ds = Dataset.from_generator_func(train_iter).repeat(config.epochs) File "/home/alphis/ws/pei_zhi_huan_jing/ERNIE/propeller/data/functional.py", line 278, in from_generator_func raise ValueError('expect generator function, got %s' % repr(_gen))

WeiyueSu commented 3 years ago

try update the propelller by python -m pip install -U paddle-propeller

HaoRan98 commented 3 years ago

@WeiyueSu Updata paddle-propeller is not making it work

WeiyueSu commented 3 years ago

It seem that you use an old version propeller in ERNIE directory. Don't use it. Try use propeller install by pip, or just comment out "/home/alphis/ws/pei_zhi_huan_jing/ERNIE/propeller/data/functional.py", line 278