在跑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删除。
问题得以解决!

本文记录了解决运行UniFormer源码时遇到的一个报错问题。错误源于create_fn函数中出现未预期的pretrained_cfg参数。通过分析timm库的版本及源码,最终发现该问题是由于库版本不匹配导致。解决方法为移除pretrained_cfg参数。
5万+





