Merge pull request NVIDIA#456 from NVIDIA/sharathts-patch-1

nvpstr · web-flow · commit af32ac2a23c5 · 2020-05-07T11:09:20.000+02:00
Fix to load Google's checkpoint
diff --git a/PyTorch/LanguageModeling/BERT/modeling.py b/PyTorch/LanguageModeling/BERT/modeling.py
@@ -106,7 +106,7 @@ def load_tf_weights_in_bert(model, tf_checkpoint_path):
         if m_name[-11:] == '_embeddings':
             pointer = getattr(pointer, 'weight')
         elif m_name == 'kernel':
-            array = np.transpose(array)
+            array = np.ascontiguousarray(np.transpose(array))
         try:
             assert pointer.shape == array.shape
         except AssertionError as e:
