在跑UniFormer源码的时候报错了,去GitHub上的issue也没人提问,自己解决后记录一下解决bug的过程。
首先直接看报错信息:
在这里插入图片描述
这里提示create_fn函数中有一个未预料到的参数pretrained_cfg,再去看UniFormer的源码中创建模型的函数:
在这里插入图片描述
可以看出作者并没有给出pretrained_cfg参数
再看anaconda3/envs/uniformer/lib/python3.7/site-packages/timm/models中create_model设置的参数,确实是有pretrained_cfg这个参数的
在这里插入图片描述
报错来源于这个函数pretrained_cfg参数:
在这里插入图片描述
猜测:很可能是timm版本的问题,但是作者并没有给出具体的版本。
解决方案:直接把这个pretrained_cfg=pretrained_cfg删除。

问题得以解决!
在这里插入图片描述

Logo

鸿蒙生态一站式服务平台。

更多推荐