Browse Source

fix agnn group

develop/0.4/predevelop
wondergo2017 4 years ago
parent
commit
bdcb6c6ee0
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      autogl/module/nas/algorithm/agnn_rl.py

+ 1
- 1
autogl/module/nas/algorithm/agnn_rl.py View File

@@ -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(


Loading…
Cancel
Save