Python TIMM包下的VIT模型报错

我最近从github上下载了MAE的模型 （https://github.com/IcarusWizard/MAE）

我按照要求安装了requirments里面的包 然后运行MAE主体程序的时候遇到了以下问题

报错的原因似乎是MAE需求安装的TIMM这个包里的VIT这个文件里面BLOCK class下面没有drop_path1这个attribute

然后我找到安装的TIMM的包

找到BLOCK这个class后好像并没有问题

请问有人遇到相似的情况么 有没有能帮忙解决一下 看看问题在哪

P.S. 我把Drop_path1和Drop_path2都改成了Drop_path之后（当然我知道这是不对的 我只是试试），又爆错没有ls1这个attribute了

好像这个class根本就不work？

该回答引用GPT
根据您提供的信息，可能是您安装的TIMM版本不兼容MAE模型所需的版本。建议您尝试更新TIMM到最新版本，或者尝试使用MAE模型作者提供的特定版本的TIMM。

另外，您修改Drop_path1和Drop_path2为Drop_path并不是正确的解决方法，因为这样可能会导致其他错误。建议您还原这些修改。

如果更新TIMM后仍然存在问题，建议您联系MAE模型的作者或者在相关论坛上寻求帮助，以获得更详细的解决方案。

这是Torch的版本原因吧，下支持gpu的
就去官网找对应于你的cuda版本的Torch，官网有安装命令的