未验证 提交 69ccb4c4 编写于 作者: Z Zhong Hui 提交者: GitHub

fix for pgl for the erniesage

fix for pgl for the erniesage
上级 5b4aef8e
# Global Enviroment Settings
# trainer config ------
n_gpu: 2 # delete it, if use cpu to train
n_gpu: 1 # number of gpus used to train, delete it, if use cpu
seed: 2020
task: "link_prediction"
......
......@@ -21,7 +21,7 @@ from paddle.io import Dataset, IterableDataset
from paddlenlp.utils.log import logger
import pgl
from pgl import Graph
from pgl.nn.sampling import graphsage_sample
from pgl.sampling import graphsage_sample
__all__ = [
"TrainData",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册