Should fix #438

19666dcb · thomwolf · 1d8c2323 · 19666dcb
Commit 19666dcb authored Apr 03, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

pytorch_pretrained_bert/modeling.py pytorch_pretrained_bert/modeling.py +2 -0

No files found.
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -91,6 +91,8 @@ def load_tf_weights_in_bert(model, tf_checkpoint_path):
                pointer = getattr(pointer, 'bias')
            elif l[0] == 'output_weights':
                pointer = getattr(pointer, 'weight')
+            elif l[0] == 'squad':
+                pointer = getattr(pointer, 'classifier')
            else:
                try:
                    pointer = getattr(pointer, l[0])