add constraints back in AE search

ekut-es · Dec 9, 2024 · e2e6555 · e2e6555
1 parent 5bd9124
commit e2e6555
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hannah/conf/nas/aging_evolution_nas.yaml b/hannah/conf/nas/aging_evolution_nas.yaml
@@ -31,6 +31,7 @@ n_jobs: 10
 presample: False
 total_candidates: 50
 num_selected_candidates: 20
+constrained_sampling_on_search: True
 bounds:
     val_error: 0.1
     # total_macs: 128000000

diff --git a/hannah/nas/search/search.py b/hannah/nas/search/search.py
@@ -248,7 +248,7 @@ def sample_candidates(
         num_candidates=None,
         sort_key="val_error",
         presample=False,
-        constrain=False,
+        constrain=True,
     ):
         candidates = []
         skip_ct = 0