MIC-DKFZ / MedNeXt

[MICCAI 2023] MedNeXt is a fully ConvNeXt architecture for 3D medical image segmentation.
https://arxiv.org/pdf/2303.09975
Apache License 2.0
345 stars 26 forks source link

Tensor size error in MedNextv1 trainer #34

Open mariaperezf opened 2 weeks ago

mariaperezf commented 2 weeks ago

Hello, I am having trouble with the following Runtime Error:

RuntimeError: The size of tensor a (3) must match the size of tensor b (4) at non-singleton dimension 2

When trying to train a model using MedNext with the command:

mednextv1_train 3d_fullres nnUNetTrainerV2_MedNeXt_S_kernel3 999 0