auto

google · Feb 23, 2017 · 0e526ca · 0e526ca
1 parent 8368971
commit 0e526ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/frame_level_models.py b/frame_level_models.py
@@ -77,7 +77,7 @@ def create_model(self, model_input, vocab_size, num_frames, **unused_params):
 
     output = slim.fully_connected(
         avg_pooled, vocab_size, activation_fn=tf.nn.sigmoid,
-        weights_regularizer=slim.l2_regularizer(1e-5))
+        weights_regularizer=slim.l2_regularizer(1e-8))
     return {"predictions": output}
 
 class DBoFModel(models.BaseModel):

diff --git a/train.py b/train.py
@@ -172,7 +172,7 @@ def build_graph(reader,
                 batch_size=1000,
                 base_learning_rate=0.01,
                 optimizer_class=tf.train.AdamOptimizer,
-                regularization_penalty=1e-3,
+                regularization_penalty=1,
                 num_readers=1,
                 num_epochs=None):
   """Creates the Tensorflow graph.