Ugrás a fő tartalomhoz

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.