Przejdź do głównej zawartości

utils

Narzędzia dla pakietu AlphaCube.

Ten moduł dostarcza wspólne narzędzia używane w całym pakiecie AlphaCube, w tym:

  • Wstępnie skonfigurowany logger rich do stylizowanego wyjścia konsolowego.
  • Automatyczne wykrywanie najlepszego dostępnego torch.device (CUDA, MPS lub CPU) i odpowiadającego mu torch.dtype.
  • Funkcje narzędziowe do kontrolowania szczegółowości (set_verbose), listowania dostępnych modeli (list_models) i definiowania globalnego katalogu pamięci podręcznej.

set_verbose

def set_verbose(loglevel=20)

Ustawia poziom szczegółowości logera.

Argumenty:

  • loglevel int - Poziom logowania (np. logging.INFO/20, logging.DEBUG/10) do kontrolowania szczegółowości.

Zwraca:

  • None

list_models

def list_models()

Wyświetla listę dostępnych identyfikatorów modeli.

Zwraca:

  • list - Lista dostępnych identyfikatorów modeli.