mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 08:56:35 +00:00
56017f2c23
update train.py with yolov10 fix generation of data.yaml
308 B
308 B
- Install dependencies first
python -m pip install --upgrade pip
pip install --upgrade huggingface_hub
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install -q git+https://github.com/THU-MIG/yolov10.git
pip install albumentations