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 dutorch.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.