Project-MONAI / research-contributions

Implementations of recent research prototypes/demonstrations using MONAI.
https://monai.io/
Apache License 2.0
1.02k stars 334 forks source link

How should I reproduce the training process of Swin UNETR on BTCV so that it can achieve the mean dice score recorded in the paper? #302

Open JeMing-creater opened 1 year ago

JeMing-creater commented 1 year ago

Describe the bug Using multi-GPU training from scratch, the BTCV segmentation score of 91.8% the average dice score cannot be achieved.

To Reproduce Steps to reproduce the behavior:

  1. Parallel training is performed on 4 T100s.
  2. The batch size is set to 1, and max_epochs is set to 5000. Other settings are selected as default.

My confusion is:
I cannot reproduce the average dice score of 91.8%, and the highest is only 83%. What training method should be used to achieve the scores recorded in the paper?

ASCI02 commented 6 months ago

Hi, have you solved this problem