mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 10:36:35 +00:00
56017f2c23
update train.py with yolov10 fix generation of data.yaml
8 lines
308 B
Markdown
8 lines
308 B
Markdown
1. 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
|
|
``` |