hitachinsk / SAMed

The implementation of the technical report: "Customized Segment Anything Model for Medical Image Segmentation"
MIT License
468 stars 43 forks source link

如何选择最优模型? #40

Open 21-10-4 opened 6 months ago

21-10-4 commented 6 months ago

我在train.py中,看到间隔几个epoch保存模型,但是如何选择最优模型?是把保存下来的模型依次跑一遍test.py,选择最好的吗

hitachinsk commented 6 months ago

其实不是,为了公平比较,基本上保存的模型是最后的模型。其实模型只要收敛就行了,训练到后面的话我们发现多训练几次少训练几次对性能差距影响可能很小。