model loading apex modification

d23eed85 · thomwolf · 1cbb32a5 · 1df6f262 · d23eed85
Commit d23eed85 authored Dec 13, 2018 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/modeling.py pytorch_pretrained_bert/modeling.py +1 -1

No files found.
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -523,7 +523,7 @@ class PreTrainedBertModel(nn.Module):
                old_keys.append(key)
                new_keys.append(new_key)
        for old_key, new_key in zip(old_keys, new_keys):
-            state_dict[new_key]=state_dict.pop(old_key)
+            state_dict[new_key] = state_dict.pop(old_key)
        missing_keys = []
        unexpected_keys = []