From 9a2d22efc978be00ab36c827a40986e38703503e Mon Sep 17 00:00:00 2001 From: wondergo2017 Date: Tue, 9 Nov 2021 08:18:53 +0000 Subject: [PATCH] resolve conflict --- test/nas/nclf_dgl.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/test/nas/nclf_dgl.py b/test/nas/nclf_dgl.py index 8e0b049..c30aeef 100644 --- a/test/nas/nclf_dgl.py +++ b/test/nas/nclf_dgl.py @@ -5,7 +5,7 @@ logging.basicConfig(level=logging.INFO) sys.path.append("../../") print(os.getcwd()) -os.environ["AUTOGL_BACKEND"] = "dgl" +os.environ["AUTOGL_BACKEND"] = "pyg" from autogl.backend import DependentBackend import dgl from dgl.data import CoraGraphDataset, CiteseerGraphDataset, PubmedGraphDataset @@ -17,15 +17,14 @@ from autogl.module.nas.space.graph_nas_macro import GraphNasMacroNodeClassificat from autogl.module.nas.estimator.one_shot import OneShotEstimator from autogl.module.nas.estimator.train_scratch import TrainEstimator from autogl.module.nas.backend import bk_feat, bk_label -from autogl.module.nas.algorithm import Darts,RL,GraphNasRL,Enas,RandomSearch,Spos +from autogl.module.nas.algorithm import Darts,RL,GraphNasRL,Enas,RandomSearch +from autogl.module.nas.algorithm import Spos from pdb import set_trace import numpy as np from autogl.datasets import build_dataset_from_name from autogl.solver.utils import set_seed set_seed(202106) -import nni.retiarii.strategy as strategy - if __name__ == '__main__': isdgl=DependentBackend.is_dgl()