Skip to content

Commit de23e5d

Browse files
authored
update xfails for PyTorch DistilBERT (huggingface#355)
1 parent 6438bce commit de23e5d

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

tank/pytorch/torch_model_list.csv

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ microsoft/MiniLM-L12-H384-uncased,True,hf,True,66M,"nlp;bert-variant;transformer
33
albert-base-v2,True,hf,True,11M,"nlp;bert-variant;transformer-encoder","12 layers; 128 embedding dim; 768 hidden dim; 12 attention heads; Smaller than BERTbase (11M params vs 109M params); Uses weight sharing to reduce # params but computational cost is similar to BERT."
44
bert-base-uncased,True,hf,True,109M,"nlp;bert-variant;transformer-encoder","12 layers; 768 hidden; 12 attention heads"
55
bert-base-cased,True,hf,True,109M,"nlp;bert-variant;transformer-encoder","12 layers; 768 hidden; 12 attention heads"
6-
distilbert-base-uncased,True,hf,True,66M,"nlp;bert-variant;transformer-encoder","Smaller and faster than BERT with 97percent retained accuracy."
76
google/mobilebert-uncased,True,hf,True,25M,"nlp,bert-variant,transformer-encoder,mobile","24 layers, 512 hidden size, 128 embedding"
87
alexnet,False,vision,True,61M,"cnn,parallel-layers","The CNN that revolutionized computer vision (move away from hand-crafted features to neural networks),10 years old now and probably no longer used in prod."
98
resnet18,False,vision,True,11M,"cnn,image-classification,residuals,resnet-variant","1 7x7 conv2d and the rest are 3x3 conv2d"

tank/test_models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,11 @@ def test_module(self, dynamic, device, config):
259259
self.module_tester.tf32 = self.pytestconfig.getoption("tf32")
260260
self.module_tester.ci = self.pytestconfig.getoption("ci")
261261
self.module_tester.ci_sha = self.pytestconfig.getoption("ci_sha")
262+
if (
263+
config["model_name"] == "distilbert-base-uncased"
264+
and config["framework"] == "torch"
265+
):
266+
pytest.xfail(reason="https://github.com/nod-ai/SHARK/issues/354")
262267
if (
263268
config["model_name"] == "facebook/convnext-tiny-224"
264269
and device == "cuda"

0 commit comments

Comments
 (0)