Swin transformer复现遇到的问题及解决方法
1)问题:TypeError: SwinTransformer: init() got an unexpected keyword argument ‘embed_dim’
2)出现原因
这个问题复现的时候师兄师姐帮我看了好久,自己也搞了好几个星期,最后师姐帮我发现问题并解决。
如下图

在不同目录下它寻找的mmdet包不同
在swin_master/(你的项目)目录下它会寻找该项目内本来包含的mmdet文件夹
但是在swin_master/tools文件夹下,它会寻找openmmlab环境中的mmdet
3)解决方法
在终端输入
export PYTHONPATH=/home/techart/xyz/swin/swin_master
# (包含mmdet文件夹的你的项目路径)
设置环境变量
指定它去该目录下寻找mmdet包
这样即使使用单GPU也可以测试
本文详细阐述了在尝试复现SwinTransformer过程中遇到的TypeError,即关于'embed_dim'的意外参数。问题源于不同目录下mmdet包查找路径不一致。解决方法是通过设置环境变量将PythonPATH指向包含mmdet文件夹的项目路径,确保单GPU测试顺利进行。
1336

被折叠的 条评论
为什么被折叠?



