Method save_pretrained

semantic_sam/BaseModel.py:22–23

(self, save_path)

        return outputs

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

    def from_pretrained(self, load_dir):
        state_dict = torch.load(load_dir, map_location='cpu')