38 lines
653 B
Markdown
38 lines
653 B
Markdown
|
# P02_errlens_training - 环境准备指南
|
||
|
|
|
||
|
|
## 依赖要求
|
||
|
|
- Python >= 3.10
|
||
|
|
- PyTorch >= 2.0
|
||
|
|
- CUDA >= 11.8 (GPU训练)
|
||
|
|
|
||
|
|
## 安装步骤
|
||
|
|
```bash
|
||
|
|
# 创建虚拟环境
|
||
|
|
python -m venv venv
|
||
|
|
source venv/bin/activate
|
||
|
|
|
||
|
|
# 安装依赖
|
||
|
|
pip install -r requirements.txt
|
||
|
|
|
||
|
|
# 下载预训练模型
|
||
|
|
python scripts/download_model.py
|
||
|
|
```
|
||
|
|
|
||
|
|
## 环境变量
|
||
|
|
| 变量名 | 说明 | 默认值 |
|
||
|
|
|--------|------|--------|
|
||
|
|
| MODEL_DIR | 模型存储目录 | ./models |
|
||
|
|
| DATA_DIR | 数据集目录 | ./data |
|
||
|
|
| GPU_ENABLED | 是否使用GPU | true |
|
||
|
|
|
||
|
|
## 运行命令
|
||
|
|
```bash
|
||
|
|
# 训练模型
|
||
|
|
python train.py
|
||
|
|
|
||
|
|
# 评估模型
|
||
|
|
python evaluate.py
|
||
|
|
|
||
|
|
# 运行测试
|
||
|
|
pytest tests/
|
||
|
|
```
|