search

Jun-Jie-Shi / M2FTrans

[IEEE-JBHI'2023] M2FTrans: Modality-Masked Fusion Transformer for Incomplete Multi-Modality Brain Tumor Segmentation
https://doi.org/10.1109/JBHI.2023.3326151
MIT License
15 stars 2 forks source link

对于val验证数据集的一个小疑惑 #3

Closed WDYTX closed 5 months ago

WDYTX commented 5 months ago

您好，我想问下是否使用了验证集（也就是val.txt）文件所指的那些数据？文章里的结果是用验证集得到的结果还是测试集得到的结果呢？感谢！！！

WDYTX commented 5 months ago

好像验证集没有用上啊

Jun-Jie-Shi commented 5 months ago

@WDYTX 您好，这里验证集的划分是为了保证与先前工作的一致性，文章的结果是用测试集得到的，当然我们也尝试了选择部分缺失情况在训练过程中进行验证(在train函数中被注释掉了)，但观察到模型1000轮次的结果似乎也尚未达到完全收敛，因此简单起见就直接选取了最后一轮的测试结果。

WDYTX commented 5 months ago

好的，非常感谢您的回复！！！

WDYTX commented 5 months ago

您好，我想问一下这个模型训练需要训练多久啊？

Jun-Jie-Shi commented 5 months ago

@WDYTX 如果设置iter_per_epoch=150的话，跑1k轮大约51 h，如果设置为每轮遍历整个数据集的话就得视数据集大小以及总轮次而变化了