search

YichiZhang98 / DTML

[PRCV 2021] The official code for "Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation"
24 stars 2 forks source link

cannot reproduce the results #3

Closed Huiimin5 closed 3 years ago

Huiimin5 commented 3 years ago

Hi, thank you so much for releasing your code!

The best performance I can obtain after training with your code on LA dataset is 0.89697391 0.81447848 7.44489285 1.85291552. I have tried all iterations saved through the training process. Do you have any idea why this performance drop comes from? By the way, may I know your python, pytorch and cuda versions?

YichiZhang98 commented 3 years ago

Hi, Thanks for your interest in our work. Our implementation is based on CUDA 10.2, python 3.8 and torch 1.7.0. In our work, the experiments are performed on the same machine several times and the results are consistent under the same setting. Possibly, the results may vary slightly due to the difference of machine/environments.