c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

embeddings resizing + tie_weights

Browse Source
This commit is contained in:
thomwolf committed 2019-07-12 00:02:49 +02:00
1 parent 50e62a4cb4
commit bd404735a7
15 files changed
+196 -332
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

pytorch_transformers/tests/modeling_xlnet_test.py
+1 -1
View File
@@ -28,7 +28,7 @@ import torch
from pytorch_transformers import (XLNetConfig, XLNetModel, XLNetLMHeadModel, XLNetForSequenceClassification, XLNetForQuestionAnswering)
from pytorch_transformers.modeling_xlnet import XLNET_PRETRAINED_MODEL_ARCHIVE_MAP
from .modeling_tests_commons import ConfigTester, create_and_check_commons, ids_tensor
from .modeling_common_test import ConfigTester, create_and_check_commons, ids_tensor
class XLNetModelTest(unittest.TestCase):
class XLNetModelTester(object):