RRT-MIL复现记录

提供数据标签的label.csv文件内容格式为：两列，第一列记录slide_id,第二列记录label，不需要表头。形如此图。

超参数含义：

--dataset_root:给定数据集根目录。根目录下需要有label.csv文件，以及特征数据集.pt文件夹。

--model_path:输出路径。输出训练后的模型权重ckp.pt文件，和最高性能权重。

--project：项目名称。与model_path路径拼接，将输出结果保存在对应项目名称文件夹内。

--model:模型名称。提供模型名称，令代码运行指定模型。

报错解决：

当模型选择clam_sb时，运行到验证的时候，会报错表示标签数据应该为tensor而不应该为None:

问题是原来代码里，验证和测试的时候，对于除了dsmil的其他模型，传入参数只有bag，并没有传入clam模型需要的标签数据。

本人将其添加了下面代码来解决报错：