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(