咨询预训练权重更换不同光谱带数据集如何修改

[lzy-coder]

使用仓库提供的模型预训练权重，可以在ICVL,CAVE数据集直接测试（因为光谱带都是31），但是如果换到Real HSI（光谱带=34）和Urban（光谱带=210）数据集进行测试，就会显示在TransformerBlock中的attn尺寸不对，修改TransformerBlock中的attn为数据集对应的光谱带数量时，就会显示预训练权重参数的尺寸对应不上，请问这个怎么解决

[Zeqiang-Lai]

HSDT/hsdt/attention.py

Line 50 in d7cca63

def __init__(self, channel, num_bands, flex=False):

把flex 改成true就可以了

[lzy-coder]

修改了GSSA中的flex为True还是不行，在Real HSI数据集
报这个错误RuntimeError: shape '[1, 31, 31]' is invalid for input of size 1156

[Zeqiang-Lai]

net.eval() 一下试试？

[lzy-coder]

这个是有的   

…

------------------ 原始邮件 ------------------ 发件人: "Zeqiang ***@***.***>; 发送时间: 2024年2月10日(星期六) 晚上10:32 收件人: ***@***.***>; 抄送: ***@***.***>; ***@***.***>; 主题: Re: [Zeqiang-Lai/HSDT] 咨询预训练权重更换不同光谱带数据集如何修改 (Issue #1) net.eval() 一下试试？ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>