Note: WiderFace scores appear low because the dataset includes extremely small faces (crowds, distant people). For kiosk use (single person close to camera), detection is highly reliable.
Total Training Cost
Model
GPU
Time
Cost
Hand pose (100 epochs)
H200 NVL (143GB)
1h 45min
~$6
Face detection (50 epochs)
H200 NVL (143GB)
54min
~$3
Total
2h 39min
~$9
Quick Start
Download Pre-trained Models
bash
1# Clone with Git LFS2git clone https://github.com/YOUR_USERNAME/yolo26-training.git
3cd yolo26-training
45# Models are in models/ (tracked with Git LFS)6ls -lh models/
Export .pt to ONNX
bash
1pip install ultralytics onnx onnxslim
23# Exports both FP32 and FP16 versions4bash scripts/export_onnx.sh checkpoints/yolo26_hand_pose.pt
5bash scripts/export_onnx.sh checkpoints/yolo26_face.pt
1pip install ultralytics onnxruntime-gpu onnx onnxslim
23# Hand pose (~1h45 on H200, ~3h on A40)4bash scripts/train_hand_pose.sh
56# Face detection (~54min on H200, ~2h on A40)7bash scripts/train_face_detect.sh
With Docker
bash
1docker build -t yolo26-training -f docker/Dockerfile .2docker run --gpus all -v $(pwd)/models:/workspace/output yolo26-training
Recommended GPUs
GPU
VRAM
Hand Pose
Face Detect
Total Cost
H200 NVL
143 GB
1h 45min
54min
~$9
H100 SXM
80 GB
~2h 30min
~1h 15min
~$10
A100
80 GB
~3h
~1h 30min
~$6
A40
48 GB
~4h
~2h
~$3
Training Tips
Hand pose: batch=512 works on 48GB+ VRAM. Lower to 128 on 24GB GPUs.
Face detect: batch=64 recommended. WiderFace has 100+ faces/image — higher batch causes OOM.
Early stopping: Both scripts use patience to stop early if metrics plateau.
WiderFace download: Script auto-downloads from HuggingFace CUHK-CSE mirror (Google Drive links are unreliable).