utils
Segédfüggvények az AlphaCube csomaghoz.
Ez a modul az AlphaCube csomagban használt általános segédfüggvényeket tartalmazza, többek között:
- Egy előre konfigurált
rich
naplózót a stílusos konzol kimenethez. - A legjobb elérhető
torch.device
(CUDA, MPS vagy CPU) és a hozzá tartozótorch.dtype
automatikus felismerése. - Segédfüggvényeket a részletesség szabályozásához (
set_verbose
), az elérhető modellek listázásához (list_models
), és a globális gyorsítótár könyvtár meghatározásához.
set_verbose
def set_verbose(loglevel=20)
Beállítja a naplózó részletességi szintjét.
Argumentumok:
loglevel
int - Naplózási szint (pl.logging.INFO
/20
,logging.DEBUG
/10
) a részletesség szabályozásához.
Visszatérési érték:
None
list_models
def list_models()
Listázza az elérhető modellazonosítókat.
Visszatérési érték:
list
- Az elérhető modellazonosítók listája.