Skip to content
Snippets Groups Projects
Commit 92267bd7 authored by tgupta6's avatar tgupta6
Browse files

lower learning rate for obj/atr graph and word vectors

parent eee09063
No related branches found
No related tags found
No related merge requests found
...@@ -742,7 +742,7 @@ class attach_optimizer(): ...@@ -742,7 +742,7 @@ class attach_optimizer():
with graph.tf_graph.as_default(): with graph.tf_graph.as_default():
all_trainable_vars = tf.trainable_variables() all_trainable_vars = tf.trainable_variables()
self.not_to_train = []#graph.object_attribute_vars + graph.word_vec_vars self.not_to_train = graph.object_attribute_vars + graph.word_vec_vars
vars_to_train = [ vars_to_train = [
var for var in all_trainable_vars var for var in all_trainable_vars
...@@ -766,9 +766,9 @@ class attach_optimizer(): ...@@ -766,9 +766,9 @@ class attach_optimizer():
self.optimizer = multi_rate_train.MultiRateOptimizer( self.optimizer = multi_rate_train.MultiRateOptimizer(
tf.train.AdamOptimizer) tf.train.AdamOptimizer)
# self.optimizer.add_variables( self.optimizer.add_variables(
# self.graph.object_attribute_vars + self.graph.word_vec_vars, self.graph.object_attribute_vars + self.graph.word_vec_vars,
# learning_rate = 1.0*self.lr) learning_rate = 0.1*self.learning_rate)
self.optimizer.add_variables( self.optimizer.add_variables(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment