pytorch版本的timm库

timm库

  • 官网文档
  • huggingface文档介绍
  • github
  • timm 是一个 PyTorch 原生实现的计算机视觉模型库。它提供了预训练模型和各种网络组件,可以用于各种计算机视觉任务,例如图像分类、物体检测、语义分割等等。
  • VIT模型
    pytorch版本的timm库_第1张图片

你可能感兴趣的:(pytorch,人工智能,python)