wjjmario

Results 1 comments of wjjmario

在models/sam.py的267行,有bug的这里会把batch_size变成1 267:masks = masks[0] 改成: 267:masks = torch.squeeze(masks,dim=1) 这样就不会把batch_size挤压掉了