Python TFElectraModel.from_pretrainedの例

コード例 #1

 def test_model_from_pretrained(self):
     # for model_name in TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
     for model_name in ["google/electra-small-discriminator"]:
         model = TFElectraModel.from_pretrained(model_name)
         self.assertIsNotNone(model)

コード例 #2

ファイル: test_modeling_tf_electra.py プロジェクト: PlusLabNLP/StoryGenerationDemo

 def test_model_from_pretrained(self):
     # for model_name in list(TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
     for model_name in ["electra-small-discriminator"]:
         model = TFElectraModel.from_pretrained(model_name,
                                                cache_dir=CACHE_DIR)
         self.assertIsNotNone(model)