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

预训练权重加载问题 #1

Closed GrandMountFoxc closed 8 months ago

GrandMountFoxc commented 8 months ago

您好，请问GitHub上提供的预训练权重应该怎么加载，直接加载会出错，会存在参数不匹配的问题。不清楚是不是我哪里设置的不对，麻烦您有时间看一下。 image image

Jun-Jie-Shi commented 8 months ago

@GrandMountFoxc 看上去您使用的模型的basic_dims应该是8，但给予的预训练权重的basic_dim应该为16；另外所给模型的输入3D裁剪大小为808080，您好像改成了128128128，从而导致了该报错。

GrandMountFoxc commented 8 months ago

好的，感谢回复