解决TypeError: SwinTransformer: __init__() got an unexpected keyword argument ‘embed_dim‘

本文详细阐述了在尝试复现SwinTransformer过程中遇到的TypeError,即关于'embed_dim'的意外参数。问题源于不同目录下mmdet包查找路径不一致。解决方法是通过设置环境变量将PythonPATH指向包含mmdet文件夹的项目路径,确保单GPU测试顺利进行。

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也可以测试

评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值