Python XxxModel.from_pretrained 예제들

예제 #1

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

예제 #2

파일: modeling_xxx_test.py 프로젝트: ddouy/transformers

 def test_model_from_pretrained(self):
     for model_name in list(XXX_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
         model = XxxModel.from_pretrained(model_name, cache_dir=CACHE_DIR)
         self.assertIsNotNone(model)