.CKPT ファイル拡張子

PyTorch Lightningは、Metaが開発した人気の機械学習フレームワークであるPyTorchのインターフェイスとして機能するPythonライブラリです。開発者は、これを使ってディープラーニングの実験を実行し、スケーラブルなディープラーニングモデルを作成できます。

モデルの作成中、開発者はそのモデルの現在の状態をCKPT、つまりチェックポイントファイルとして保存できます。これにより、開発者は将来、そのバージョンのモデルに戻って使用できます。各CKPTファイルには、そのバージョンのモデルを読み込むために必要なすべての情報が含まれています。これには、現在のエポック、グローバルステップ、LightningModuleのstate_dictなどが含まれます。

CKPTファイルは開くことを目的としていません。PyTorch Lightningのマルチプラットフォームモデルとして読み込み、使用することを目的としています。たとえば、CKPTファイルに含まれるモデルをPyTorch Lightningで読み込むには、次のコマンドを使用します。

model = MyLightningModule.load_from_checkpoint("/Path/To/YourFile.ckpt")