Mark big downloads slow (#7325)

* Make big downloads as slow

* Add import

* Right order for slow decorator

* More slow tests
This commit is contained in:
Sylvain Gugger
2020-09-22 12:21:52 -04:00
committed by GitHub
parent 585217c87f
commit 1ee2194fb6
3 changed files with 22 additions and 11 deletions

View File

@@ -4,7 +4,7 @@ import datasets
import numpy as np
from transformers import AutoTokenizer, TrainingArguments, is_torch_available
from transformers.testing_utils import get_tests_dir, require_torch
from transformers.testing_utils import get_tests_dir, require_torch, slow
if is_torch_available():
@@ -287,6 +287,7 @@ class TrainerIntegrationTest(unittest.TestCase):
trainer.train()
self.check_trained_model(trainer.model, alternate_seed=True)
@slow
def test_trainer_eval_mrpc(self):
MODEL_ID = "bert-base-cased-finetuned-mrpc"
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
@@ -301,6 +302,7 @@ class TrainerIntegrationTest(unittest.TestCase):
result = trainer.evaluate()
self.assertLess(result["eval_loss"], 0.2)
@slow
def test_trainer_eval_lm(self):
MODEL_ID = "distilroberta-base"
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)