- 博客(4)
- 收藏
- 关注
原创 cannot import name ‘XXX‘ from ‘XXX‘解决思路
首先确保自己已经安装过timm库,然后找到本地目录(我这里pycharm解释器是在anaconda虚拟环境下的,因此是该目录。你的具体目录可以从报错信息中找到):XXX/ProgramData/Anaconda3/envs/XXX/Lib/site-packages/timm/models/layers/打开init.py文件,查找报错的weight_init。
2024-01-05 10:07:20
4443
3
原创 (深度学习入门)如何根据官方代码创建模型(以fasternet为例)
接下来在models文件夹下打开fasternet.py文件,可以看到该文件中只有关于FasterNet的定义,但是没有关于FasterNet_S模型的调用函数,因此我们这里需要新写一个FasterNet_S函数来方便后续的模型创建。如果文件中没提到该参数的值,就直接参照FasterNet的init函数中的值写即可。接下来打开解压缩后的项目文件,可以看到在cfg目录下有不同类型模型对应的参数详情,这里fasternet_s对应的是fasternet_s.yaml文件,打开后可以看到里面写的模型参数配置。
2024-01-04 17:00:13
655
2
原创 使用pth文件创建model
情况一:如果该模型在timm中存在,则可以将下载好的pth文件放到XX/user/.cache/torch/hub/checkpoints目录下。②翻墙去huggingface里搜索。①Git开源项目里会有下载链接;
2024-01-04 16:40:23
691
1
原创 no module named XXX from timm解决方案
在使用时,第一个参数为文件名,将pretrained设为false(这样就不会再从网络上下载模型文件,直接使用本地checkpoints中的)。比如下载maxvit_tiny_rw_224(rw是对应pytorch使用的版本,tf则是tensorflow)。然后将该文件放到XXX/.cache/torch/hub/checkpoints目录下。翻墙去huggingface的timm中搜索对应预训练模型文件。下载pytorch_model.bin文件即可。
2024-01-02 20:17:27
491
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人