如何选择DeepLung代码测试时的权重文件（.ckpt）？

Question

使用DeepLung代码完成LUNA16胸腔CT图片数据集的训练后，在测试上有疑问

完成测试后，每个数据集（总共10个）产生了100个参数文件或权重文件（.ckpt），如下图：

一个帖子给出的测试代码如下（帖子选择的权重文件是100.ckpt）：

python main.py --model dpn3d26 -b 1 --resume results/dpn3d26/retrft960/100.ckpt --test 1 --save-dir dpn3d26/retrft960/ --config config_training0

那么，在测试的时候，不知如何选择这些权重文件（.ckpt）。应该选择其中1个或几个，还是需要把这100个权重文件（.ckpt）全部都挨个测试一遍吗？

请帮忙解答该问题，并简单解释关于如何选择权重文件的思路。

Answer 1

在测试模型时，通常只使用一个权重文件，并且是在训练过程中最优的权重文件。如果要选择最优的权重文件，可以在训练时保存不同迭代次数的权重文件，然后在验证集上评估这些权重文件，选择性能最好的权重文件进行测试。这种方法称为"模型选择"。

如果没有验证集，可以在训练集上按照一定的间隔（例如每隔10个迭代次数）保存权重文件，然后在训练集上评估这些权重文件，选择性能最好的权重文件进行测试。这种方法称为"交叉验证"。

可以使用训练集或验证集来选择最优的权重文件，但是最终的测试结果应该是在测试集上得到的。

对于这个问题，如果已经有了验证集或训练集，可以使用上述方法选择最优的权重文件；如果没有，可以选择在训练集上性能最好的权重文件进行测试。不需要把所有的100个权重文件全部测试一遍。
仅供参考，望采纳，谢谢。