d8f9592b209e368e4934b6b429443c2a8673de3f,tests/models/test_mmbt.py,TestMMBTTorchscript,test_finetune_model,#TestMMBTTorchscript#,40

Before 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):

After Change

        test_sample.image = torch.rand((3, 300, 300)).float()
        test_sample_list = SampleList([test_sample.copy()])

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

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

Instances