search

Dootmaan / MT-UNet

Official Code for *Mixed Transformer UNet for Medical Image Segmentation*
MIT License
160 stars 26 forks source link

Bug? #19

Closed Chenguang-Wang closed 2 years ago

Chenguang-Wang commented 2 years ago

I think this is a bug in the train_mtunet_ACDC.py code. Could you example it? 88 val_outputs = model(val_image_batch) 89 val_outputs = torch.argmax(torch.softmax(outputs, dim=1), dim=1).squeeze(0) It should be: 88 val_outputs = model(val_image_batch) 89 val_outputs = torch.argmax(torch.softmax(val_outputs , dim=1), dim=1).squeeze(0)

Dootmaan commented 2 years ago

Thank you @Chenguang-Wang. This bug is caused by the fixing of issue #13 . We have corrected it in both branches now.