# vision_transformer

**Repository Path**: olivegame/vision_transformer

## Basic Information

- **Project Name**: vision_transformer
- **Description**: vision-transformer 代码注释
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-09-28
- **Last Updated**: 2025-09-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# vision_transformer

#### 介绍
vision-transformer 代码注释

#### 效果
1. 以CWRU数据集的小样本数据进行10分类，采用“vit_base_patch16_224.pth”作为预训练集，运行 train1000.py 结果如下：
![训练结果](optimized_training_curves.png)

2. 如果是使用源代码的flower数据应该使用“vit_base_patch16_224_in21k.pth”作为预训练集,运行 train.py