timmdocs:Ross Wightman的Timm图像模型库的文档

目录 pytorch图像模型（timm） timm是由创建的深度学习库，它是SOTA计算机视觉模型，层，实用程序，优化器，调度程序，数据加载器，增强功能以​​及具有再现ImageNet训练结果能力的训练/验证脚本的集合。 安装 pip install timm 或者对于可编辑的安装， git clone https://github.com/rwightman/pytorch-image-models cd pytorch-image-models && pip install -e . 如何使用 建立模型 import timm import torch model = timm . create_model ( 'resnet34' ) x = torch . randn ( 1 , 3 , 224 , 224 ) model ( x ). shape 使用timm创建