Pular para o conteúdo principal

utils

Utilitários para o Pacote AlphaCube.

Este módulo fornece utilitários comuns usados em todo o pacote AlphaCube, incluindo:

  • Um logger rich pré-configurado para saída de console estilizada.
  • Detecção automática do melhor torch.device disponível (CUDA, MPS ou CPU) e do torch.dtype correspondente.
  • Funções utilitárias para controlar a verbosidade (set_verbose), listar modelos disponíveis (list_models) e definir o diretório de cache global.

set_verbose​

def set_verbose(loglevel=20)

Define o nível de verbosidade do logger.

Argumentos:

  • loglevel int - Nível de log (ex: logging.INFO/20, logging.DEBUG/10) para controlar a verbosidade.

Retorna:

  • None

list_models​

def list_models()

Lista os IDs dos modelos disponíveis.

Retorna:

  • list - Uma lista dos IDs dos modelos disponíveis.