SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

remove derived classes for now

Browse Source
This commit is contained in:
thomwolf
2019-08-05 19:08:19 +02:00
parent 13936a9621
commit 0b524b0848
4 changed files with 4 additions and 295 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pytorch_transformers/tests/modeling_auto_test.py
View File

@@ -21,7 +21,7 @@ import shutil
import pytest
import logging
from pytorch_transformers import AutoConfig, BertConfig, AutoModel, BertModel, AutoModelForSequenceClassification, AutoModelWithLMHead
from pytorch_transformers import AutoConfig, BertConfig, AutoModel, BertModel
from pytorch_transformers.modeling_bert import BERT_PRETRAINED_MODEL_ARCHIVE_MAP
from .modeling_common_test import (CommonTestCases, ConfigTester, ids_tensor)
@@ -42,14 +42,6 @@ class AutoModelTest(unittest.TestCase):
for value in loading_info.values():
self.assertEqual(len(value), 0)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
self.assertIsNotNone(model)
self.assertIsInstance(getattr(model, model.base_model_prefix), BertModel)
model = AutoModelWithLMHead.from_pretrained(model_name)
self.assertIsNotNone(model)
self.assertIsInstance(getattr(model, model.base_model_prefix), BertModel)
if __name__ == "__main__":
unittest.main()