j028

nmcardoso · Aug 23, 2023 · daae27d · daae27d
1 parent 9d179e0
commit daae27d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mergernet/jobs/j028.py b/mergernet/jobs/j028.py
@@ -30,6 +30,7 @@ def call(self):
       HP.const('pretrained_weights', 'imagenet'),
       HP.const('batch_size', 128),
       HP.const('dense_1_units', 400),
+      HP.const('batch_norm_1', True),
       HP.const('dropout_1_rate', 0.32),
       # metrics and validation
       HP.const('metrics', ['f1', 'recall', 'roc']),