Zum Hauptinhalt springen

utils

Hilfsfunktionen für das AlphaCube-Paket.

Dieses Modul stellt allgemeine Hilfsfunktionen bereit, die im gesamten AlphaCube-Paket verwendet werden, darunter:

  • Ein vorkonfigurierter rich-Logger für formatierte Konsolenausgaben.
  • Automatische Erkennung des besten verfügbaren torch.device (CUDA, MPS oder CPU) und des entsprechenden torch.dtype.
  • Hilfsfunktionen zur Steuerung der Ausführlichkeit (set_verbose), zum Auflisten verfügbarer Modelle (list_models) und zur Definition des globalen Cache-Verzeichnisses.

set_verbose

def set_verbose(loglevel=20)

Setzt die Ausführlichkeitsstufe des Loggers.

Argumente:

  • loglevel int - Logging-Level (z. B. logging.INFO/20, logging.DEBUG/10) zur Steuerung der Ausführlichkeit.

Rückgabewert:

  • None

list_models

def list_models()

Listet die verfügbaren Modell-IDs auf.

Rückgabewert:

  • list - Eine Liste der verfügbaren Modell-IDs.