2024年8月11日
摘要: 简要介绍 timm库,全称pytorch-image-models,是最前沿的PyTorch图像模型、预训练权重和实用脚本的开源集合库,其中的模型可用于训练、推理和验证。 github源码链接: https://github.com/huggingface/pytorch-image-models 阅读全文
posted @ 2024-08-11 21:35 零度的python武器库 阅读(901) 评论(0) 推荐(0) 编辑
摘要: 运行代码: import timm import torch model = timm.create_model( 'deit_small_patch16_224', pretrained=True, num_classes=6, pretrained_cfg_overlay = dict(file 阅读全文
posted @ 2024-08-11 10:57 零度的python武器库 阅读(436) 评论(0) 推荐(0) 编辑