j027

nmcardoso · Aug 19, 2023 · f073105 · f073105
1 parent 4095a1d
commit f073105
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mergernet/jobs/j027.py b/mergernet/jobs/j027.py
@@ -27,7 +27,6 @@ def call(self):
     hps = HyperParameterSet(
       HP.const('architecture', 'efficientnetv2b0'),
       HP.const('pretrained_weights', 'imagenet'),
-      HP.const('batch_size', 64),
       # HP.const('metrics', ['tpr']),
       HP.const('positive_class_id', 1),
       HP.const('negative_class_id', 0),
@@ -40,8 +39,9 @@ def call(self):
       HP.num('lr_decay_steps', low=0.5, high=0.9),
       HP.num('lr_decay_alpha', low=0.1, high=1.0),
       HP.num('opt_lr', low=1e-5, high=1e-3, log=True),
-      HP.num('weight_decay', low=1e-4, high=1e-1, log=True),
+      HP.num('weight_decay', low=1e-4, high=1e-1),
       HP.num('label_smoothing', low=0, high=0.1),
+      HP.num('batch_size', low=64, high=256, step=64, 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, step=1, dtype=int),