def test_finetune_model(self): model = self.finetune_model.eval() self.assertTrue( test_utils.compare_torchscript_transformer_models( model, vocab_size=BERT_VOCAB_SIZE ) )
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))
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))