Ejemplos de TFT5Model.from_pretrained en Python

Ejemplo n.º 1

 def test_model_from_pretrained(self):
     for model_name in ["t5-small"]:
         model = TFT5Model.from_pretrained(model_name, cache_dir=CACHE_DIR)
         self.assertIsNotNone(model)

Ejemplo n.º 2

Archivo: modeling_tf_t5_test.py Proyecto: zxlzr/AwesomeMRC

 def test_model_from_pretrained(self):
     cache_dir = "/tmp/transformers_test/"
     for model_name in ['t5-small']:
         model = TFT5Model.from_pretrained(model_name, cache_dir=cache_dir)
         shutil.rmtree(cache_dir)
         self.assertIsNotNone(model)