Update modules.py

fix as_list function call bug
Kyubyong · Dec 25, 2018 · d2b0281 · d2b0281
1 parent 6672f93
commit d2b0281
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules.py b/modules.py
@@ -195,7 +195,7 @@ def multihead_attention(queries,
     with tf.variable_scope(scope, reuse=reuse):
         # Set the fall back option for num_units
         if num_units is None:
-            num_units = queries.get_shape().as_list[-1]
+            num_units = queries.get_shape().as_list()[-1]
 
         # Linear projections
         Q = tf.layers.dense(queries, num_units, activation=tf.nn.relu) # (N, T_q, C)