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

[lukaszbinden]

found a fix: I commented out line 47 in
data/lidc_dataset.py:

# y = y.transpose(2,0,1)

now it's training.

[lukaszbinden]

for reference, I was able to train and test the model on LIDC, and got these results:

2023-05-17 09:19:01,785 INFO Running Experiment: MoSE_run
2023-05-17 09:19:07,087 INFO model size: 41.60 / MB
2023-05-17 09:19:07,608 INFO Loading model ./logs/lidc/MoSE_run/MoSE_run_best_ged.pth
2023-05-17 09:20:27,032 INFO - Mean GED: 0.210
2023-05-17 09:20:27,033 INFO - Mean M-IoU: 0.621
2023-05-17 09:20:27,033 INFO - Mean ECE: 0.083%

[gaozhitong]

Hi, Lukas. Thank you very much for your feedback on the bug, I will fix it later. I apologize for the late response since I am really busy last week preparing my NeurIPS submission. Feel free to contact me if there is any other problem.