Aller au contenu principal

utils

Utilitaires pour le paquet AlphaCube.

Ce module fournit des utilitaires communs utilisés dans l'ensemble du paquet AlphaCube, notamment :

  • Un logger rich préconfiguré pour une sortie console stylisée.
  • La détection automatique du meilleur torch.device disponible (CUDA, MPS ou CPU) et du torch.dtype correspondant.
  • Des fonctions utilitaires pour contrôler la verbosité (set_verbose), lister les modèles disponibles (list_models) et définir le répertoire de cache global.

set_verbose

def set_verbose(loglevel=20)

Définit le niveau de verbosité du logger.

Arguments:

  • loglevel int - Niveau de journalisation (par ex., logging.INFO/20, logging.DEBUG/10) pour contrôler la verbosité.

Retourne:

  • None

list_models

def list_models()

Liste les identifiants des modèles disponibles.

Retourne:

  • list - Une liste des identifiants des modèles disponibles.