本文介绍了使用VOC格式数据集训练PoolFormer模型时遇到的两个常见错误及其解决方案。一是模型输入通道数不匹配导致的运行时错误，解决方法是确保模型的in_channels参数与数据集相匹配；二是优化器adamw引发的错误，通过修复PyTorch源代码中对应的bug可以解决。
使用voc格式数据集训练poolformer 时两个报
报错1：
return torch._C._nn.linear(input, weight, bias)
RuntimeError: mat1 dim 1 must match mat2 dim 0解决办法是模型的in_channels要与模型本身匹配，我开始使用的是2048，就报这个错了，改成768后就正常了
报错2：
由于使用的优化器是adamw，查了资料后发现，是pytorch代码bug导致的，github上已经修复
local variable ‘beta1‘ referenced before assignment将pytorch 的代码按照github上的对应代码文件修改一下即可
总共涉及三个文件：
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
