Open 21-10-4 opened 6 months ago
我在train.py中,看到间隔几个epoch保存模型,但是如何选择最优模型?是把保存下来的模型依次跑一遍test.py,选择最好的吗
其实不是,为了公平比较,基本上保存的模型是最后的模型。其实模型只要收敛就行了,训练到后面的话我们发现多训练几次少训练几次对性能差距影响可能很小。
我在train.py中,看到间隔几个epoch保存模型,但是如何选择最优模型?是把保存下来的模型依次跑一遍test.py,选择最好的吗