NLUT icon indicating copy to clipboard operation
NLUT copied to clipboard

Unexpected key(s) in state_dict

Open unihornWwan opened this issue 1 year ago • 0 comments

加载下载的预训练模型时报错 Unexpected key(s) in state_dict: "blurer.op.1.weight", "SB1.conv1.conv2d.weight", "SB1.conv1.conv2d.bias", "SB1.conv1.bn.weight", "SB1.conv1.bn.bias", "SB1.conv1.bn.running_mean", "SB1.conv1.bn.running_var", "SB1.conv1.bn.num_batches_tracked", "SB1.conv2.conv2d.weight", "SB1.conv2.conv2d.bias", "SB1.conv2.bn.weight", "SB1.conv2.bn.bias", "SB1.conv2.bn.running_mean", "SB1.conv2.bn.running_var", "SB1.conv2.bn.num_batches_tracked". ,经检查 NLUTNet 模型定义时确实没有 blurer 和 SB1 相关的定义,辛苦作者 check 下代码和参数文件,谢谢!

unihornWwan avatar Dec 11 '23 06:12 unihornWwan