added tests + fixed losses

2019-01-08 16:24:23 +01:00
parent eed51c5bdf
commit 3cf12b235a
4 changed files with 484 additions and 225 deletions
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -549,7 +549,7 @@ class BertPreTrainedModel(nn.Module):
                model.__class__.__name__, unexpected_keys))
        if len(error_msgs) > 0:
            raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
-                               self.__class__.__name__, "\n\t".join(error_msgs)))
+                               model.__class__.__name__, "\n\t".join(error_msgs)))
        if tempdir:
            # Clean up temp dir
            shutil.rmtree(tempdir)