From 0e526caace96d3cf6f0686757d568f9ffba998b4 Mon Sep 17 00:00:00 2001 From: Nisarg Kothari Date: Wed, 22 Feb 2017 16:22:45 -0800 Subject: [PATCH] auto --- frame_level_models.py | 2 +- train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/frame_level_models.py b/frame_level_models.py index a18e6509..15b74a2e 100644 --- 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 index 05851127..e4cd05ca 100644 --- 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.