From bebccfa19e40295230cdc7dc09de65e19a44d5aa Mon Sep 17 00:00:00 2001 From: Frozenmad Date: Tue, 19 Oct 2021 02:11:21 +0000 Subject: [PATCH] fix bugs of pyg backend --- autogl/module/train/graph_classification_full.py | 1 + 1 file changed, 1 insertion(+) diff --git a/autogl/module/train/graph_classification_full.py b/autogl/module/train/graph_classification_full.py index 6ebcb8b..fa045a0 100644 --- a/autogl/module/train/graph_classification_full.py +++ b/autogl/module/train/graph_classification_full.py @@ -314,6 +314,7 @@ class GraphClassificationFullTrainer(BaseGraphClassificationTrainer): if self.pyg_dgl == 'pyg': data = data.to(self.device) pred.append(self.model.model(data)) + label.append(data.y) elif self.pyg_dgl == 'dgl': data = [data[i].to(self.device) for i in range(len(data))] _, labels = data