Перейти к основному содержимому

utils

Утилиты для пакета AlphaCube.

Этот модуль предоставляет общие утилиты, используемые во всем пакете AlphaCube, включая:

  • Предварительно настроенный логгер rich для стилизованного вывода в консоль.
  • Автоматическое определение наилучшего доступного torch.device (CUDA, MPS или CPU) и соответствующего torch.dtype.
  • Утилитарные функции для управления подробностью вывода (set_verbose), перечисления доступных моделей (list_models) и определения глобального каталога кэша.

set_verbose

def set_verbose(loglevel=20)

Устанавливает уровень подробности логгера.

Аргументы:

  • loglevel int - Уровень логирования (например, logging.INFO/20, logging.DEBUG/10) для управления подробностью вывода.

Возвращает:

  • None

list_models

def list_models()

Перечисляет доступные идентификаторы моделей.

Возвращает:

  • list - Список доступных идентификаторов моделей.