search

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
393 stars 12 forks source link

Size of input error #17

Open oliea1 opened 3 months ago

oliea1 commented 3 months ago

when i start training the error occur my input_size_h = 256 input_size_w = 256 ValueError: Using a target size (torch.Size([4, 65536])) that is different to the input size (torch.Size([4, 196608])) is deprecated. Please ensure they have the same size.

JCruan519 commented 3 months ago

@oliea1 Hello! Could you please provide the specific location of the error?

CollinsCl7 commented 2 months ago

when i start training the error occur my input_size_h = 256 input_size_w = 256 ValueError: Using a target size (torch.Size([4, 65536])) that is different to the input size (torch.Size([4, 196608])) is deprecated. Please ensure they have the same size.

hello,did you solve the problem?I have the same error

oliea1 commented 2 months ago

when i start training the error occur my input_size_h = 256 input_size_w = 256 ValueError: Using a target size (torch.Size([4, 65536])) that is different to the input size (torch.Size([4, 196608])) is deprecated. Please ensure they have the same size.

hello,did you solve the problem?I have the same error

Not yet