7ce17a58e7b61b1bc2fc7384c1974e60967bd9fa,tests/models/test_visual_bert.py,TestVisualBertTorchscript,test_finetune_model,#TestVisualBertTorchscript#,89

Before Change

        self.finetune_model.model.eval()
        input_ids = torch.randint(low=0, high=30255, size=(1, 128)).long()
        input_mask = torch.ones((1, 128)).long()
        attention_mask = torch.ones((1, 228)).long()
        token_type_ids = torch.zeros(1, 128).long()
        visual_embeddings = torch.rand((1, 100, 2048)).float()
        visual_embeddings_type = torch.zeros(1, 100).long()

After Change

    @test_utils.skip_if_no_network
    def test_finetune_model(self):
        self.finetune_model.eval()
        sample_list = SampleList()

        sample_list.add_field(
            "input_ids", torch.randint(low=0, high=30255, size=(1, 128)).long()
        )
        sample_list.add_field("input_mask", torch.ones((1, 128)).long())
        sample_list.add_field("segment_ids", torch.zeros(1, 128).long())
        sample_list.add_field("image_feature_0", torch.rand((1, 100, 2048)).float())

        with torch.no_grad():
            model_output = self.finetune_model(sample_list)

Instances