add readme

README.md

-# vision-transformers-cifar10
-Let's train vision transformers for cifar 10! 
-
-This is an unofficial and elementary implementation of `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale`.
-
-I use pytorch for implementation.
-
-### Updates
-* Added [ConvMixer]((https://openreview.net/forum?id=TVHS5Y4dNvM)) implementation. Really simple! (2021/10)
-
-* Added wandb train log to reproduce results. (2022/3)
-
-* Added CaiT and ViT-small. (2022/3)
-
-* Added SwinTransformers. (2022/3)
-
-* Added MLP mixer. (2022/6)
-
-* Changed default training settings for ViT.
-
-# Usage example
-`python train_cifar10.py` # vit-patchsize-4
-
-`python train_cifar10.py  --size 48` # vit-patchsize-4-imsize-48
-
-`python train_cifar10.py --patch 2` # vit-patchsize-2
-
-`python train_cifar10.py --net vit_small --n_epochs 400` # vit-small
-
-`python train_cifar10.py --net vit_timm` # train with pretrained vit
-
-`python train_cifar10.py --net convmixer --n_epochs 400` # train with convmixer
-
-`python train_cifar10.py --net mlpmixer --n_epochs 500 --aug --lr 1e-3`
-
-`python train_cifar10.py --net cait --n_epochs 200` # train with cait
-
-`python train_cifar10.py --net swin --n_epochs 400` # train with SwinTransformers
-
-`python train_cifar10.py --net res18` # resnet18+randaug
-
-# Results..
-
-|             | Accuracy | Train Log |
-|:-----------:|:--------:|:--------:|
-| ViT patch=2 |    80%    | |
-| ViT patch=4 Epoch@200 |    80%   | [Log](https://wandb.ai/arutema47/cifar10-challange/reports/Untitled-Report--VmlldzoxNjU3MTU2?accessToken=3y3ib62e8b9ed2m2zb22dze8955fwuhljl5l4po1d5a3u9b7yzek1tz7a0d4i57r) |
-| ViT patch=4 Epoch@500 |    88%   | [Log](https://wandb.ai/arutema47/cifar10-challange/reports/Untitled-Report--VmlldzoxNjU3MTU2?accessToken=3y3ib62e8b9ed2m2zb22dze8955fwuhljl5l4po1d5a3u9b7yzek1tz7a0d4i57r) |
-| ViT patch=8 |    30%   | |
-| ViT small  | 80% | |
-| MLP mixer |    88%   | |
-| CaiT  | 80% | |
-| Swin-t  | 90% | |
-| ViT small (timm transfer) | 97.5% | |
-| ViT base (timm transfer) | 98.5% | |
-| [ConvMixerTiny(no pretrain)](https://openreview.net/forum?id=TVHS5Y4dNvM) | 96.3% |[Log](https://wandb.ai/arutema47/cifar10-challange/reports/convmixer--VmlldzoyMjEyOTk1?accessToken=2w9nox10so11ixf7t0imdhxq1rf1ftgzyax4r9h896iekm2byfifz3b7hkv3klrt)|
-|   resnet18  |  93%  | |
-|   resnet18+randaug  |  95%  | [Log](https://wandb.ai/arutema47/cifar10-challange/reports/Untitled-Report--VmlldzoxNjU3MTYz?accessToken=968duvoqt6xq7ep75ob0yppkzbxd0q03gxy2apytryv04a84xvj8ysdfvdaakij2) |
-
-# Used in..
-* Vision Transformer Pruning [arxiv](https://arxiv.org/abs/2104.08500) [github](https://github.com/Cydia2018/ViT-cifar10-pruning)
+# Vision-Transformers-cifar10_PyTorch
+
+## 模型介绍
+
+Vision Transformers (ViT) 是一种基于 Transformer 的视觉模型，它在图像分类任务上表现出了与当前最先进的卷积神经网络相当的性能。而ViT模型是使用Transformer架构来代替传统卷积神经网络的。ViT模型的核心思想是将输入的图像分割成数个图块（patch），然后将这些图块拼接成一个序列，再输入到Transformer中进行特征提取和分类。
+
+## 模型结构
+
+Vision Transformers (ViT) - CIFAR10 模型的结构基于 Transformer 的视觉模型，其核心思想是将输入的图像分割成数个图块（patch），然后将这些图块拼接成一个序列，再输入到 Transformer 中进行特征提取和分类。
+
+具体来说，该模型的输入是一张 $32\times32$ 的彩色图像，其中每个图像被分割成 $4\times4$ 个大小为 $8\times8$ 的图块。这些图块被重塑为一个序列，并传递给 Transformer 编码器进行特征提取。在编码器中，每个序列元素（即每个图块）都经过一个多头自注意力机制（Multi-Head Self-Attention）模块和一个前馈网络模块（Feed-Forward Network）进行处理。这些模块的输出被送入下一层编码器，直到最终输出分类结果。
+
+该模型的主要组成部分是由多个 Transformer 编码器层组成的。每个编码器层由一个多头自注意力机制模块和一个前馈网络模块组成。这些编码器层之间使用残差连接（Residual Connection）连接，以便有效地传播信息和减轻梯度消失问题。
+
+最后，模型的输出通过一个全连接层进行分类，该层将 Transformer 编码器的输出展平为一维张量，并对其进行分类。整个模型的训练过程使用交叉熵损失函数进行优化。
+
+## 数据集
+
+本次使用的数据集为[CIFAR-10](https://www.cs.toronto.edu/~kriz/cifar.html)，CIFAR-10数据集由10类中的60000 32x32颜色图像组成，每个类别有6000张图像。有50000个训练图像和10000个测试图像。
+
+该数据集分为五个训练批次和一批测试批次，每个测试批次有10000张图像。该测试批次完全包含来自每个类的1000个随机选择的图像。训练批次包含剩余的图像，但有些训练批次可能包含一个班级的图像，而不是另一个班级的图像。在他们之间，训练批次包含每个班级的5000张图像。
+
+## 训练及推理
+
+### 环境配置
+
+提供光源拉取的训练镜像
+
+```
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py38-latest
+```
+
+安装依赖：
+
+```
+pip install pandas==1.5.3
+```
+
+创建软链接
+
+```
+ln -s datasets/cifar10 data
+```
+
+若需要指定路径，也可以在工程根目录的train_cifar10.py中修改（修改130行左右的trainset和testset变量即可）
+
+开始训练
+
+```
+bash train.sh
+#默认使用四卡训练，可在train.sh脚本中修改
+```
+
+## 准确率数据
+
+| 卡数 | 准确率 |
+| :--: | :----: |
+|  4   | 84.91% |
+
+
+
+## 源码仓库及问题反馈
+
+https://developer.hpccube.com/codes/modelzoo/vision-transformers-cifar10_pytorch
+
+## 参考
+
+https://github.com/kentaroy47/vision-transformers-cifar10
\ No newline at end of file

model.properties

+# 模型名称
+modelName= Vision-Transformers-cifar10_PyTorch
+# 模型描述
+modelDescription=Vision Transformers (ViT) 是一种基于 Transformer 的视觉模型
+# 应用场景(多个标签以英文逗号分割)
+appScenario=训练,pytorch,transformer
+# 框架类型(多个标签以英文逗号分割)
+frameType=PyTorch