提交 4a0e30ed 编写于 作者: Y yelrose

Add linkprediction for ogb

上级 f88dc512
......@@ -132,7 +132,6 @@ def train_and_evaluate(exe,
ret = metric.parse(ret)
if global_step % train_log_step == 0:
sys.stderr.write(json.dumps(ret) + '\n')
for key, value in ret.items():
writer.add_scalar(
'train_' + key, value, global_step=global_step)
......
......@@ -30,15 +30,15 @@ ssl._create_default_https_context = ssl._create_unverified_context
import numpy as np
import multiprocessing
import pgl
import paddle
import paddle.fluid as F
import paddle.fluid.layers as L
import pgl
from args import parser
from utils.args import print_arguments, check_cuda
from utils.init import init_checkpoint, init_pretraining_params
from args import parser
from model import BaseGraph, GCNGraph
from model import BaseGraph
from dataloader.ogbl_ppa_dataloader import PPADataGenerator
from monitor.train_monitor import train_and_evaluate, OgbEvaluator
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册