AmorFati2016

> I have the same qustion ``` sam = sam_model_registry["vit_h"](checkpoint=src_path) torch.save(sam.prompt_encoder.state_dict(),'weights/sam_vit_h_prompt_encoder.pth') torch.save(sam.mask_decoder.state_dict(),'weights/sam_vit_h_mask_decoder.pth') ```

> > > I have the same qustion > > > > > > ``` > > sam = sam_model_registry["vit_h"](checkpoint=src_path) > > torch.save(sam.prompt_encoder.state_dict(),'weights/sam_vit_h_prompt_encoder.pth') > > torch.save(sam.mask_decoder.state_dict(),'weights/sam_vit_h_mask_decoder.pth') > > ``` >...