update all tf.shape and tensor.shape to shape_list

2019-11-28 15:51:43 +01:00
parent 1ab8dc44b3
commit adb5c79ff2
13 changed files with 48 additions and 54 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -118,7 +118,7 @@ if is_torch_available():

 # TensorFlow
 if is_tf_available():
-    from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, TFSequenceSummary
+    from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, TFSequenceSummary, shape_list
    from .modeling_tf_auto import (TFAutoModel, TFAutoModelForSequenceClassification, TFAutoModelForQuestionAnswering,
                                   TFAutoModelWithLMHead)