Fix and returning to Adam as default

matsengrp · Jun 12, 2024 · 960359f · 960359f
1 parent d5783fd
commit 960359f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/netam/framework.py b/netam/framework.py
@@ -372,7 +372,7 @@ def __init__(
         train_dataset,
         val_dataset,
         model,
-        optimizer_name="AdamW",
+        optimizer_name="Adam",
         batch_size=1024,
         learning_rate=0.1,
         min_learning_rate=1e-4,
@@ -757,6 +757,7 @@ def __init__(
         train_dataset,
         val_dataset,
         model,
+        optimizer_name="Adam",
         batch_size=1024,
         learning_rate=0.1,
         min_learning_rate=1e-4,
@@ -767,11 +768,12 @@ def __init__(
             train_dataset,
             val_dataset,
             model,
-            batch_size,
-            learning_rate,
-            min_learning_rate,
-            l2_regularization_coeff,
-            name,
+            optimizer_name=optimizer_name,
+            batch_size=batch_size,
+            learning_rate=learning_rate,
+            min_learning_rate=min_learning_rate,
+            l2_regularization_coeff=l2_regularization_coeff,
+            name=name,
         )
 
     def loss_of_batch(self, batch):