Remove references to removed l1 penalty

HazyResearch · Sep 7, 2018 · c215a87 · c215a87
1 parent f25849e
commit c215a87
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/metal/end_model/em_defaults.py b/metal/end_model/em_defaults.py
@@ -25,7 +25,6 @@
         # Train Loop
         "n_epochs": 10,
         # 'grad_clip': 0.0,
-        "l1": 0.0,
         "l2": 0.0,
         "validation_metric": "accuracy",
         "validation_freq": 1,

diff --git a/metal/tuners/tuner.py b/metal/tuners/tuner.py
@@ -112,7 +112,7 @@ def config_generator(search_space, max_search, shuffle=True):
             search_space = {
                 'verbose': True,                              # discrete
                 'n_epochs': 100,                              # discrete
-                'l1': [0.0, 0.01, 0.1],                       # discrete
+                'momentum': [0.0, 0.9, 0.99],                       # discrete
                 'l2': {'range': [0.0001, 10]}                 # linear range
                 'lr': {'range': [0.001, 1], 'scale': 'log'},  # log range
             }