TransUNet icon indicating copy to clipboard operation
TransUNet copied to clipboard

not converge with custom dataset

Open Chq2018515 opened this issue 2 years ago • 0 comments

thanks for your great work! I was using pretrained model R50-ViT-B_16 and trained it on some opensource dataset for lung segmentation. During training, the total loss stayed above 0.7 and ce loss stayed at 0.3. And in testing phase, i got results like "mean_dice : 0.354955 mean_hd95 : 77.297413 mean_precision : 0.373490". I wonder what i can do with it. Any help would be appreciated : )

Chq2018515 avatar Mar 17 '22 11:03 Chq2018515