Python TFAlbertModel.from_pretrainedの例

コード例 #1

ファイル: modeling_tf_albert_test.py プロジェクト: MichaelCaohn/Prune-Testing

 def test_model_from_pretrained(self):
     cache_dir = "/tmp/transformers_test/"
     # for model_name in list(TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
     for model_name in ['albert-base-uncased']:
         model = TFAlbertModel.from_pretrained(
             model_name, cache_dir=cache_dir)
         shutil.rmtree(cache_dir)
         self.assertIsNotNone(model)

コード例 #2

ファイル: test_modeling_tf_albert.py プロジェクト: yunjae2/transformers

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

コード例 #3

ファイル: test_modeling_tf_albert.py プロジェクト: donatasrep/pytorch-pretrained-BERT

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