ValuError: Target size (torch.Size([14, 254, 254])) must be the same as input size (torch.Size([8000, 10, 14, 14]))
I am building a custom model for instance segmentation task that takes in 2D images, and a target for training. I think there is an issue with the model arch but not sure where it is. I have tried many things and used diff ways to extract features but stock on similar errors like above.
ValueError: Target size (torch.Size([14, 254, 254])) must be the same as input size (torch.Size([8000, 10, 14, 14]))
I am building a custom model for instance segmentation task that takes in 2D images, and a target for training. I think there is an issue with the model arch but not sure where it is. I have tried many things and used diff ways to extract features but stock on similar errors like above.