ai_soc_sw/projects/P02_errlens_training/ENVIRONMENT.md

P02_errlens_training - 环境准备指南

依赖要求

• Python >= 3.10
• PyTorch >= 2.0
• CUDA >= 11.8 (GPU训练)

安装步骤

# 创建虚拟环境
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

运行命令

# 训练模型
python train.py

# 评估模型
python evaluate.py

# 运行测试
pytest tests/