Saltar al contenido principal

utils

Utilidades para el Paquete AlphaCube.

Este módulo proporciona utilidades comunes utilizadas en todo el paquete AlphaCube, incluyendo:

  • Un registrador rich preconfigurado para una salida de consola con estilo.
  • Detección automática del mejor torch.device disponible (CUDA, MPS o CPU) y el torch.dtype correspondiente.
  • Funciones de utilidad para controlar la verbosidad (set_verbose), listar los modelos disponibles (list_models) y definir el directorio de caché global.

set_verbose​

def set_verbose(loglevel=20)

Establece el nivel de verbosidad del registrador.

Argumentos:

  • loglevel int - Nivel de registro (p. ej., logging.INFO/20, logging.DEBUG/10) para controlar la verbosidad.

Retorna:

  • None

list_models​

def list_models()

Lista los IDs de los modelos disponibles.

Retorna:

  • list - Una lista de los IDs de los modelos disponibles.