diff --git a/ogb_examples/linkproppred/ogbl-ppa/monitor/train_monitor.py b/ogb_examples/linkproppred/ogbl-ppa/monitor/train_monitor.py index fb118c914ef0a7b91ab5951be6951f84cf2dda33..a517b7c2679f51f4247912df8f661a20792720b8 100644 --- a/ogb_examples/linkproppred/ogbl-ppa/monitor/train_monitor.py +++ b/ogb_examples/linkproppred/ogbl-ppa/monitor/train_monitor.py @@ -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) diff --git a/ogb_examples/linkproppred/ogbl-ppa/train.py b/ogb_examples/linkproppred/ogbl-ppa/train.py index eecf8ae7c842fc428d7468385010a0b719e2f18d..c70fa4f9dd4987e615f6f935b5108c727fe7abee 100644 --- a/ogb_examples/linkproppred/ogbl-ppa/train.py +++ b/ogb_examples/linkproppred/ogbl-ppa/train.py @@ -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