Should fix #438

2019-04-03 11:01:01 +02:00
parent 1d8c232324
commit 19666dcb3b
1 changed files with 2 additions and 0 deletions
--- 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])