xujinzh / xujinzh.github.io

个人博客主页
https://xujinzh.github.io/
3 stars 2 forks source link

使用 timm 模块查看常见深度学习模型结构 | J. Xu #345

Open xujinzh opened 10 months ago

xujinzh commented 10 months ago

https://xujinzh.github.io/2023/11/24/python-look-model-structure-by-timm/?

深度学习中预训练模型库非常重要,它能够帮助我们非常方便的获取到模型的结构、模型的权重文件等,这大大降低了入门深度学习的门槛,如高性能的硬件设备(服务器、GPU),同时使用迁移学习的思想能够大大缩短我们开发可实用模型的时间。常见的预训练模型库包含有 torchvision.models(CV 模型)、transformers(CV 和 NLP 大模型相关)、timm(包含 CV 领域小模型和大模型,