From b271df41c3f0e7e67eff5fb8e3601a7c97dfc0f9 Mon Sep 17 00:00:00 2001 From: lihy96 Date: Fri, 15 Oct 2021 12:15:26 +0800 Subject: [PATCH] node --- autogl/module/train/node_classification_full.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autogl/module/train/node_classification_full.py b/autogl/module/train/node_classification_full.py index 0a56131..103b6d1 100644 --- a/autogl/module/train/node_classification_full.py +++ b/autogl/module/train/node_classification_full.py @@ -229,7 +229,7 @@ class NodeClassificationFullTrainer(BaseNodeClassificationTrainer): if hasattr(data, "val_mask") and data.val_mask is not None: if type(self.feval) is list: feval = self.feval[0] - else: + else:branch feval = self.feval val_loss = self.evaluate([data], mask=data.val_mask, feval=feval) if feval.is_higher_better() is True: