mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 08:46:36 +00:00
56017f2c2307d5d9c360bdfb4cea493d274ef99c
update train.py with yolov10 fix generation of data.yaml
- 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
Description
Languages
Python
83.7%
Shell
16.3%