这个屌丝很懒，什么也没留下！

Vision Transformer 实现图片分类_vit_base_patch16_224.pth

作者：小蓝xlanll | 2024-03-22 11:34:17

vit_base_patch16_224.pth

采用vit_base_patch16_224.pth 作为预训练模型进行加载，并冻结其参数，微调全连接最后一层，要不然训练时间太长了

data_transform = {
   
    # 训练中的数据增强和归一化
    'train': transforms.Compose([
        transforms.Resize(size=(224,224))1
2
3
4

本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/287852