Model Loader
このモジュールは、学習済みのAlphaCubeソルバーモデルを読み込んで使用するための関数とクラスを提供します。
Function:
load_model(model_id, cache_dir)
: 学習済みのAlphaCubeソルバーモデルを読み込みます。
Classes:
Model
: AlphaCubeソルバーのMLPアーキテクチャです。LinearBlock
: MLPアーキテクチャの構成要素です。
load_model
def load_model(
model_id="small",
cache_dir=os.path.expanduser("~/.cache/alphacube")
)
学習済みのAlphaCubeソルバーモデルを読み込みます。
引数:
model_id
str - 読み込むモデルのバリアント("small"、"base"、または "large"