Polyp-PVT icon indicating copy to clipboard operation
Polyp-PVT copied to clipboard

最优权重

Open lsh-med opened this issue 1 year ago • 3 comments

你好,我想问问最后的最优权重是怎么选取的,是根据总的测试文件进行选取的最优吗,还是根据各个不同的测试数据集进行测试分别选取的

lsh-med avatar Apr 24 '24 12:04 lsh-med

将所有的测试集汇总,取其最优,详细实现参考https://github.com/DengPingFan/Polyp-PVT/blob/cd71fbf9fcd8c32cc0f33699f4e0186073c64dc4/Train.py#L117C9-L117C18

dongbo811 avatar Apr 26 '24 11:04 dongbo811

那考虑要训练资源有限,可以分别对Kvasir和ClinicDB的数据集进行测试保留权重,再分别验证其各自的泛化能力吗

lsh-med avatar Apr 28 '24 03:04 lsh-med

就是我的模型可能参数比较多,达不到和您一样的训练标准

lsh-med avatar Apr 28 '24 03:04 lsh-med