Merge pull request #677 from ufal/name_fix

Fixing back the variable names in the decoder
ufal · Mar 21, 2018 · 5686ac8 · 5686ac8
2 parents 46d816b + d7e8be0
commit 5686ac8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neuralmonkey/decoders/autoregressive.py b/neuralmonkey/decoders/autoregressive.py
@@ -156,15 +156,15 @@ def decoding_w(self) -> tf.Variable:
                 return tf.transpose(self.embedding_matrix)
 
             return get_variable(
-                "logit_matrix",
+                "state_to_word_W",
                 [self.output_dimension, len(self.vocabulary)],
                 initializer=tf.glorot_uniform_initializer())
 
     @tensor
     def decoding_b(self) -> tf.Variable:
         with tf.name_scope("output_projection"):
             return get_variable(
-                "logit_bias", [len(self.vocabulary)],
+                "state_to_word_b", [len(self.vocabulary)],
                 initializer=tf.zeros_initializer())
 
     @tensor