fix

nmcardoso · Aug 18, 2023 · 0dd5118 · 0dd5118
1 parent c383748
commit 0dd5118
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mergernet/core/hp.py b/mergernet/core/hp.py
@@ -130,7 +130,7 @@ def num(
     name: str,
     low: Union[float, int],
     high: Union[float, int],
-    step: Union[float, int] = 1,
+    step: Union[float, int] = None,
     log: bool = False,
     dtype: Union[float, int] = float
   ) -> Union[FloatHyperParameter, IntHyperParameter]:

diff --git a/mergernet/jobs/j026.py b/mergernet/jobs/j026.py
@@ -30,9 +30,9 @@ def call(self):
       HP.const('tl_epochs', 12),
       HP.const('epochs', 24),
       HP.const('batch_size', 64),
-      HP.num('dense_1_units', low=32, high=1024, dtype=int),
+      HP.num('dense_1_units', low=32, high=1024, step=1, dtype=int),
       HP.num('dropout_1_rate', low=0.1, high=0.5),
-      HP.num('dense_2_units', low=32, high=1024, dtype=int),
+      HP.num('dense_2_units', low=32, high=1024, step=1, dtype=int),
       HP.num('dropout_2_rate', low=0.1, high=0.5),
       HP.num('opt_lr', low=1e-5, high=1e-3, log=True)
     )