utils
Utilitaires pour le paquet AlphaCube.
Ce module fournit des utilitaires communs utilisés dans l'ensemble du paquet AlphaCube, notamment :
- Un logger
richpréconfiguré pour une sortie console stylisée. - La détection automatique du meilleur
torch.devicedisponible (CUDA, MPS ou CPU) et dutorch.dtypecorrespondant. - 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:
loglevelint - 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.