From bdcb6c6ee032e15f5a165aab4a2f5d4b7ea9a08e Mon Sep 17 00:00:00 2001 From: wondergo2017 Date: Thu, 9 Jun 2022 10:41:27 +0800 Subject: [PATCH] fix agnn group --- autogl/module/nas/algorithm/agnn_rl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autogl/module/nas/algorithm/agnn_rl.py b/autogl/module/nas/algorithm/agnn_rl.py index c15a6da..4d61817 100644 --- a/autogl/module/nas/algorithm/agnn_rl.py +++ b/autogl/module/nas/algorithm/agnn_rl.py @@ -103,7 +103,7 @@ class AGNNRL(GraphNasRL): for tag in tags: if tag in field.name: groups[tag].append(field) - groups=list(groups.values()) + groups=[x for x in groups.values() if x] # controller self.controller = AGNNActionGuider(