AIモデルの拡張子

LLM関連ででてくるモデルの拡張子。

拡張子 発生元 関連ライブラリ 概要
.pb Google TensorFlow, TensorFlow Lite SavedModel形式
Procol Bufferの略
アーキテクチャと重みを一緒に保存す
.ckpt Google TensorFlow CheckPointの略
トレーニング途中の状態を保存
.pth Meta PyTorch PyTorchの略
モデルのパラメータだけではなく、オプティマイザの状態も保存可能
.pt Meta PyTorch PyTorchの略
モデルのパラメータだけではなく、オプティマイザの状態も保存可能
.h5 HDF Grouo Keras, h5py Hierarchical Data Format version 5の略
フレームワークに依存しない
.onnx Microsft + Meta Open Neural Networks Exchangeの略
フレームワークに依存しない
.safetensors HuggingFace llama.cpp, whisper.cpp 大規模言語モデルをローカルで使用できる
? GGUF llama.cpp 大規模言語モデルをローカルで使用できる
write: 2024/06/09/ 10:00:00