# source_yolov5

**Repository Path**: computergr/source_yolov5

## Basic Information

- **Project Name**: source_yolov5
- **Description**: yolov5     
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-03-10
- **Last Updated**: 2025-08-31

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

1. 训练的主文件是main.py, enter()为训练模型的入口，需要输入5个参数送入
2. epochs是模型训练的迭代次数，可自行定义
3. batch_size是每次送入内存的样本数量，根据CPU和GPU的大小而定，一般设置为2的倍数，例如32，64，128等
4. pretrain_model是已经训练好模型的索引号，用为微调使用，可以为空
5. training_data是训练数据集的索引号
6. patience是停止模型训练的标准，当经过连续patience次后模型的评测指标依旧没有变化，模型停止训练