update optimizer path

rewiringamerica · Nov 7, 2024 · 734e5f6 · 734e5f6
1 parent 322a26a
commit 734e5f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/surrogate_model.py b/src/surrogate_model.py
@@ -210,7 +210,7 @@ def conv_batchnorm_relu(
 
         final_model.compile(
             loss=masked_mae,
-            optimizer="adam",
+            optimizer=tf.keras.optimizers.Adam()
             # metrics=[mape],
         )
         return final_model