Fix issue: #1962, input's shape seem to cause error in 2.2.0 version tf_albert_model

2019-11-27 09:58:38 +00:00
parent 45d767297a
commit 88b317739f
1 changed files with 1 additions and 1 deletions
--- a/transformers/modeling_tf_albert.py
+++ b/transformers/modeling_tf_albert.py
@@ -645,7 +645,7 @@ class TFAlbertModel(TFAlbertPreTrainedModel):
        if input_ids is not None and inputs_embeds is not None:
            raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
        elif input_ids is not None:
-            input_shape = input_ids.shape
+            input_shape = tf.shape(input_ids)
        elif inputs_embeds is not None:
            input_shape = inputs_embeds.shape[:-1]
        else: