事前訓練モデルデータを読み込むとエラーがでる
モデルを読み込むとKeyErrorが発生するということが起きたが、このモデルを用いて過去に実験を回していたので、エラーがでるはずがない。
そこで、原因を探ると、仮想環境上のlightningのバージョンがあがってしまったのが原因ではないかと推測したので、それについてまとめる。
解決策
現在のエラーが発生しているバージョンが2.4.0だった。そこで、前バージョンの2.3.3に下げてみたが、同じようなエラーが発生した(微妙に無いといわれるKeyが変わった気がしたが...)。次に2.2.5に下げたところ、正常に動作することを確認した。
まとめ
学習済みのモデルを読み込んだときにKeyErrorが発生する場合は、lightningのバージョンを下げてみると解決できるかも?