SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

reducing my level of enthousiasm

Browse Source
This commit is contained in:
thomwolf
2019-11-28 15:56:53 +01:00
committed by Lysandre Debut
parent adb5c79ff2
commit 4a666885b5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformers/modeling_tf_utils.py
View File

@@ -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):