Esempi in Python per compare_torchscript_transformer_models

Esempio n. 1

File: test_visual_bert.py Progetto: facebookresearch/mmf

 def test_finetune_model(self):
     model = self.finetune_model.eval()
     self.assertTrue(
         test_utils.compare_torchscript_transformer_models(
             model, vocab_size=BERT_VOCAB_SIZE
         )
     )

Esempio n. 2

File: test_mmf_transformer.py Progetto: SunYanCN/pythia

 def test_finetune_xlmr_base(self):
     self.config.model_config[
         self.model_name]["transformer_base"] = "xlm-roberta-base"
     model = build_model(self.config.model_config[self.model_name])
     model.eval()
     self.assertTrue(
         test_utils.compare_torchscript_transformer_models(
             model, vocab_size=XLM_ROBERTA_VOCAB_SIZE))

Esempio n. 3

File: test_mmf_transformer.py Progetto: SunYanCN/pythia

 def test_finetune_bert_base(self):
     model = build_model(self.config.model_config[self.model_name])
     model.eval()
     self.assertTrue(
         test_utils.compare_torchscript_transformer_models(
             model, vocab_size=BERT_VOCAB_SIZE))