Method forward

semantic_sam/BaseModel.py:18–20

(self, *inputs, **kwargs)

        self.model = module

    def forward(self, *inputs, **kwargs):
        outputs = self.model(*inputs, **kwargs)
        return outputs

    def save_pretrained(self, save_path):
        torch.save(self.model.state_dict(), save_path)