CWSAM icon indicating copy to clipboard operation
CWSAM copied to clipboard
verified publisher icon xypu98

about batch_size

Open wjjmario opened this issue 1 year ago • 1 comments

in models/sam.py line267，got a bug make batch_size turn to 1 267：masks = masks[0] changed： 267：masks = torch.squeeze(masks,dim=1)

wjjmario avatar Jul 19 '24 05:07 wjjmario

very good

ymkasad avatar Sep 03 '25 11:09 ymkasad