OpenMythosのサンプルプログラムを動かしました
1.概要
前回、OpenMythosの環境構築まで実現したので、GitHubにあるサンプルプログラム(Usage)を動かしてみました。一部エラーが発生しましたが、結果出力部分なので修正をしました。その内容を記述します。
2.詳細
Usageに記載されているPythonプログラムをopenmythos_usage.pyとして作成します。
$ source ~/mypy/bin/activate
(mypy) $ pythoh3 openmythos_usage.py
下記エラーとなります
[MLA] Parameters: 1,538,626
[MLA] Logits shape: torch.Size([2, 16, 1000])
[MLA] Generated shape: torch.Size([2, 24])
Traceback (most recent call last):
File "/home/nakasima/openmythos/openmythos_usage.py", line 48, in <module>
rho = torch.linalg.eigvals(A).abs().max().item()
^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: linalg.eig: The input tensor A must have at least 2 dimensions.
コードの下記部分を修正します
#rho = torch.linalg.eigvals(A).abs().max().item()
rho = A.abs().max().item()
$ source ~/mypy/bin/activate
(mypy) $ pythoh3 openmythos_usage.py
[MLA] Parameters: 1,538,626
[MLA] Logits shape: torch.Size([2, 16, 1000])
[MLA] Generated shape: torch.Size([2, 24])
[MLA] Spectral radius ρ(A) = 0.3679 (must be < 1)
動作しました。
参考
[本ブログ内参照]
・OpenMythosをLinuxMint 22.3に環境構築してみました
[外部サイト参照]
・OpenMythosがローカル環境で動くか検証してみた
コメント
コメントを投稿