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
richdo 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:
loglevelint - 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.