Browse Source

add act&concat search

tags/v0.3.1
wondergo2017 5 years ago
parent
commit
bb41925da4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      examples/test_graph_nas.py

+ 1
- 1
examples/test_graph_nas.py View File

@@ -30,7 +30,7 @@ if __name__ == '__main__':
lr_scheduler_type=None,),
nas_algorithms=[Enas(num_epochs=10)],
#nas_algorithms=[Darts(num_epochs=200)],
nas_spaces=[GraphNasNodeClassificationSpace(hidden_dim=16, ops=[GCNConv, GCNConv])],
nas_spaces=[GraphNasNodeClassificationSpace(hidden_dim=16, ops=[GCNConv, GCNConv],search_act_con=True)],
nas_estimators=[OneShotEstimator()]
)
solver.fit(dataset)


Loading…
Cancel
Save