Lewislou
/

cellseg_sribd

Image Segmentation

cell segmentation

Inference Endpoints

Model card Files Files and versions Community

Lewislou commited on Mar 22, 2023

Commit

31e9a7d

•

1 Parent(s): 3fa8784

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -83,7 +83,8 @@ for i in range(3):
 #config = ModelConfig()
 #print(config)
 my_model = MyModel.from_pretrained("Lewislou/cellseg_sribd")
 with torch.no_grad():
     output = my_model(pre_img_data)

 #config = ModelConfig()
 #print(config)
 my_model = MyModel.from_pretrained("Lewislou/cellseg_sribd")
+checkpoints = torch.load('model.pt')
+my_model.load_checkpoints(checkpoints)
 with torch.no_grad():
     output = my_model(pre_img_data)