TypeError: __init__() got an unexpected keyword argument ‘pretrained_cfg‘
在跑UniFormer源码的时候报错了,去GitHub上的issue也没人提问,自己解决后记录一下解决bug的过程。中create_model设置的参数,确实是有pretrained_cfg这个参数的。猜测:很可能是timm版本的问题,但是作者并没有给出具体的版本。可以看出作者并没有给出pretrained_cfg参数。解决方案:直接把这个。
·
在跑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
删除。
问题得以解决!
更多推荐
已为社区贡献1条内容
所有评论(0)