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

[common attributes] Slightly sharper test coverage

Browse Source
This commit is contained in:
Julien Chaumond
2019-11-11 19:56:45 -05:00
parent 9d2398fd99
commit 2f17464266
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
transformers/tests/modeling_common_test.py
View File

@@ -468,9 +468,15 @@ class CommonTestCases:
for model_class in self.all_model_classes: for model_class in self.all_model_classes:
model = model_class(config) model = model_class(config)
model.get_input_embeddings() self.assertIsInstance(
model.get_input_embeddings(),
torch.nn.Embedding
)
model.set_input_embeddings(torch.nn.Embedding(10, 10)) model.set_input_embeddings(torch.nn.Embedding(10, 10))
model.get_output_embeddings() x = model.get_output_embeddings()
self.assertTrue(
x is None or isinstance(x, torch.nn.Linear)
)
def test_tie_model_weights(self): def test_tie_model_weights(self):
if not self.test_torchscript: if not self.test_torchscript: