Esempi in Python per TF_XLM_PRETRAINED_MODEL_ARCHIVE_MAP

Esempio n. 1

 def test_model_from_pretrained(self):
     cache_dir = "/tmp/transformers_test/"
     for model_name in list(TF_XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
         model = XLMModel.from_pretrained(model_name, cache_dir=cache_dir)
         shutil.rmtree(cache_dir)
         self.assertIsNotNone(model)

Esempio n. 2

File: test_modeling_tf_xlm.py Progetto: deepchatterjeevns/pytorch-transformers

 def test_model_from_pretrained(self):
     for model_name in list(TF_XLM_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
         model = TFXLMModel.from_pretrained(model_name)
         self.assertIsNotNone(model)