Opciones Globales
Estas opciones funcionan en la mayoría de comandos del CLI de AITraining.
Versión
Verifica la versión instalada:
aitraining --version
aitraining -v
Ayuda
Obtén ayuda para cualquier comando:
aitraining --help
aitraining llm --help
aitraining text-classification --help
Archivo de Configuración
Carga parámetros desde un archivo de configuración YAML:
aitraining --config path/to/config.yaml
- Experimentos reproducibles
- Configuraciones complejas
- Compartir configuraciones con compañeros de equipo
Backend
Especifica dónde se ejecuta el entrenamiento:
aitraining llm --train --backend local ...
| Backend | Descripción |
|---|
local | Ejecutar en máquina local (predeterminado). Variantes: local-cli, local-ui |
spaces-* | Ejecutar en Hugging Face Spaces |
ep-* | Hugging Face Endpoints |
ngc-* | NVIDIA NGC/DGX Cloud |
nvcf-* | NVIDIA Cloud Functions |
Opciones de Backend Spaces
| Backend | GPU |
|---|
spaces-t4-small | T4 (small) |
spaces-t4-medium | T4 (medium) |
spaces-a10g-small | A10G (small) |
spaces-a10g-large | A10G (large) |
spaces-a10g-largex2 | 2x A10G |
spaces-a10g-largex4 | 4x A10G |
spaces-a100-large | A100 |
spaces-l4x1 | 1x L4 |
spaces-l4x4 | 4x L4 |
spaces-l40sx1 | 1x L40S |
spaces-l40sx4 | 4x L40S |
spaces-l40sx8 | 8x L40S |
spaces-cpu-basic | Solo CPU |
spaces-cpu-upgrade | CPU (mejorado) |
Los backends remotos requieren autenticación: Al usar backends no locales (spaces-*, ep-*, ngc-*, nvcf-*), debes proporcionar --username y --token para autenticación en Hugging Face.
Push a Hub también requiere autenticación: Incluso con --backend local, usar --push-to-hub requiere --username y --token para subir el modelo a Hugging Face Hub.
Variables de Entorno
Configura estas antes de ejecutar comandos:
Autenticación
export HF_TOKEN="hf_..." # Token de Hugging Face
export WANDB_API_KEY="..." # Clave de Weights & Biases
Configuración de GPU
export CUDA_VISIBLE_DEVICES=0,1 # Usar GPUs específicas
export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512 # Gestión de memoria
Control MPS (Apple Silicon)
export AUTOTRAIN_DISABLE_MPS=1 # Forzar entrenamiento CPU en Mac
export AUTOTRAIN_ENABLE_MPS=1 # Forzar MPS incluso con cuantización
Modo Interactivo
Lanza el asistente de configuración:
aitraining # Sin argumentos = modo asistente
aitraining llm --interactive # Modo interactivo explícito
Próximos Pasos
