Skip to content

GitLab

Commit 4a666885 authored by thomwolf's avatar thomwolf Committed by Lysandre Debut
Browse files

reducing my level of enthousiasm

parent adb5c79f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
transformers/modeling_tf_utils.py
View file @ 4a666885
...@@ -494,7 +494,7 @@ class TFSequenceSummary(tf.keras.layers.Layer): ...@@ -494,7 +494,7 @@ class TFSequenceSummary(tf.keras.layers.Layer):
def shape_list(x): def shape_list(x):
"""Deal with dynamic shape in tensorflow cleanly.""" """Deal with dynamic shape in tensorflow cleanly."""
static = x.shape.as_list() static = x.shape.as_list()
dynamic = shape_list(x) dynamic = tf.shape(x)
return [dynamic[i] if s is None else s for i, s in enumerate(static)] return [dynamic[i] if s is None else s for i, s in enumerate(static)]
def get_initializer(initializer_range=0.02): def get_initializer(initializer_range=0.02):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment