bd1822d9a84e157c468e78dafc8a8db20aa4c62f,tests/models/test_mmbt.py,TestMMBTTorchscript,test_finetune_model,#TestMMBTTorchscript#,47

Before Change

    @test_utils.skip_if_no_network
    def test_finetune_model(self):
        self.finetune_model.eval()
        test_sample = Sample()
        test_sample.input_ids = torch.randint(low=0, high=30255, size=(128,)).long()
        test_sample.input_mask = torch.ones(128).long()
        test_sample.segment_ids = torch.zeros(128).long()
        test_sample.image = torch.rand((3, 300, 300)).float()
        test_sample_list = SampleList([test_sample])

        with torch.no_grad():
            model_output = self.finetune_model.model(test_sample_list)

        script_model = torch.jit.script(self.finetune_model.model)
        with torch.no_grad():
            script_output = script_model(test_sample_list)

        self.assertTrue(torch.equal(model_output["scores"], script_output["scores"]))


class TestMMBTConfig(unittest.TestCase):

After Change

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


class TestMMBTConfig(unittest.TestCase):
    def test_mmbt_from_params(self):

Instances