You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thank you very much for the code, I have a question, about the abdominal multi-organ segmentation dataset - Synapse, I see that there is a train_synpse.py file in the source code, but there is no corresponding experimental result in the paper, and when it is reproduced, an error is encountered.
#----------Training----------#
Traceback (most recent call last):
File "/mnt/workspace/VM-UNetV2/train_synapse.py", line 221, in
main(config)
^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/train_synapse.py", line 158, in main
loss = train_one_epoch(
^^^^^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/engine_synapse.py", line 48, in train_one_epoch
loss = criterion(out, targets)
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pai/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/utils.py", line 344, in forward
loss_ce = F.cross_entropy(input_for_ce, target)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pai/lib/python3.11/site-packages/torch/nn/functional.py", line 3029, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: input and target batch or spatial sizes don't match: target [32, 224, 224], input [32, 56, 56, 96]
I would be very grateful if you could help me.
The text was updated successfully, but these errors were encountered:
Thank you very much for the code, I have a question, about the abdominal multi-organ segmentation dataset - Synapse, I see that there is a train_synpse.py file in the source code, but there is no corresponding experimental result in the paper, and when it is reproduced, an error is encountered.
#----------Training----------#
Traceback (most recent call last):
File "/mnt/workspace/VM-UNetV2/train_synapse.py", line 221, in
main(config)
^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/train_synapse.py", line 158, in main
loss = train_one_epoch(
^^^^^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/engine_synapse.py", line 48, in train_one_epoch
loss = criterion(out, targets)
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pai/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/workspace/VM-UNetV2/utils.py", line 344, in forward
loss_ce = F.cross_entropy(input_for_ce, target)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/pai/lib/python3.11/site-packages/torch/nn/functional.py", line 3029, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: input and target batch or spatial sizes don't match: target [32, 224, 224], input [32, 56, 56, 96]
I would be very grateful if you could help me.
The text was updated successfully, but these errors were encountered: