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