Python TFOpenAIGPTModel.from_pretrained示例

示例#1

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

示例#2

 def test_model_from_pretrained(self):
     for model_name in TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
         model = TFOpenAIGPTModel.from_pretrained(model_name)
         self.assertIsNotNone(model)

示例#3

文件： test_modeling_tf_openai_gpt.py 项目： BluePill27/transformers

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