0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

Traceback (most recent call last): File "/usr/local/lib/python3.10/dist-packages/lightning/fabric/utilities/data.py", line 487, in __getattr__ return self[key] KeyError: 'num_classes'

Posted at

事前訓練モデルデータを読み込むとエラーがでる

モデルを読み込むとKeyErrorが発生するということが起きたが、このモデルを用いて過去に実験を回していたので、エラーがでるはずがない。
そこで、原因を探ると、仮想環境上のlightningのバージョンがあがってしまったのが原因ではないかと推測したので、それについてまとめる。

解決策

現在のエラーが発生しているバージョンが2.4.0だった。そこで、前バージョンの2.3.3に下げてみたが、同じようなエラーが発生した(微妙に無いといわれるKeyが変わった気がしたが...)。次に2.2.5に下げたところ、正常に動作することを確認した。

まとめ

学習済みのモデルを読み込んだときにKeyErrorが発生する場合は、lightningのバージョンを下げてみると解決できるかも?

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?