ViT:图像值得16x16字-源码

上传者: 42140716 | 上传时间: 2021-05-14 09:13:53 | 文件大小: 78KB | 文件类型: ZIP
视觉变形金刚 在PyTorch中实现,这是一种使用变压器样式编码器在视觉分类中实现SOTA的新模型。相关文章。 特征 香草维生素 混合ViT(支持BiTResNets作为骨干网) 混合ViT(支持AxialResNets作为骨干网) 训练脚本 去做: 训练脚本 支持线性衰减 正确的超级参数 全轴向ViT Imagenet-1K和Imagenet-21K的结果 安装 创建环境: conda env create -f environment.yml 准备数据集: mkdir data cd data ln -s path/to/dataset imagenet 运行脚本 对于非分布式培训: python train.py --model ViT --name vit_logs 对于分布式培训: CUDA_VISIBLE_DEVICES=0,1,2,3 python dist_tra

文件下载

资源详情

[{"title":"( 23 个子文件 78KB ) ViT:图像值得16x16字-源码","children":[{"title":"ViT-main","children":[{"title":".gitignore <span style='color:#111;'> 1.78KB </span>","children":null,"spread":false},{"title":"dist_train.py <span style='color:#111;'> 13.10KB </span>","children":null,"spread":false},{"title":"LICENSE <span style='color:#111;'> 1.04KB </span>","children":null,"spread":false},{"title":"static","children":[{"title":"model.png <span style='color:#111;'> 53.64KB </span>","children":null,"spread":false}],"spread":true},{"title":"vit","children":[{"title":"datasets","children":[{"title":"__init__.py <span style='color:#111;'> 96B </span>","children":null,"spread":false},{"title":"cifar.py <span style='color:#111;'> 2.34KB </span>","children":null,"spread":false},{"title":"imagenet1K.py <span style='color:#111;'> 1.98KB </span>","children":null,"spread":false}],"spread":true},{"title":"__init__.py <span style='color:#111;'> 368B </span>","children":null,"spread":false},{"title":"models","children":[{"title":"__init__.py <span style='color:#111;'> 166B </span>","children":null,"spread":false},{"title":"Attention.py <span style='color:#111;'> 4.86KB </span>","children":null,"spread":false},{"title":"PositionalEncoding.py <span style='color:#111;'> 1.39KB </span>","children":null,"spread":false},{"title":"AxialNet.py <span style='color:#111;'> 7.37KB </span>","children":null,"spread":false},{"title":"ViT.py <span style='color:#111;'> 7.32KB </span>","children":null,"spread":false},{"title":"Transformer.py <span style='color:#111;'> 1.99KB </span>","children":null,"spread":false},{"title":"BiT.py <span style='color:#111;'> 4.95KB </span>","children":null,"spread":false},{"title":"HybridViT.py <span style='color:#111;'> 6.63KB </span>","children":null,"spread":false}],"spread":true},{"title":"build_optimizer.py <span style='color:#111;'> 1.22KB </span>","children":null,"spread":false},{"title":"utils.py <span style='color:#111;'> 5.33KB </span>","children":null,"spread":false},{"title":"metrics.py <span style='color:#111;'> 443B </span>","children":null,"spread":false},{"title":"build_scheduler.py <span style='color:#111;'> 2.90KB </span>","children":null,"spread":false}],"spread":true},{"title":"README.md <span style='color:#111;'> 1.81KB </span>","children":null,"spread":false},{"title":"environment.yml <span style='color:#111;'> 1.49KB </span>","children":null,"spread":false},{"title":"train.py <span style='color:#111;'> 9.00KB </span>","children":null,"spread":false}],"spread":true}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明