メインコンテンツまでスキップ

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/20logging.DEBUG/10)。

戻り値:

  • None

list_models

def list_models()

利用可能なモデルIDを一覧表示します。

戻り値:

  • list - 利用可能なモデルIDのリスト。