深度學習模型權重檔案格式與存放目錄
隨著深度學習模型的發展，越來越多的開發者透過 GitHub 與 Hugging Face 分享模型權重，以便其他人可以下載並加以應用。但不同的深度學習框架有各自的儲存格式與資料夾結構，因此了解這些規範能幫助我們更快速找到所需的模型。 1. 常見的深度學習權重檔案格式 不同的深度學習框架使用不同的檔案格式來儲存模型的權重，以下是最常見的副檔名： 副檔名 用途 對應框架 .bin PyTorch 模型權重 (Hugging Face) PyTorch .pth / .pt PyTorch 權重 (state_dict 或完整模型 ) PyTorch .safetensors 更安全的 PyTorch 權重存儲格式 PyTorch, Hugging Face .pb TensorFlow Frozen Graph TensorFlow .ckpt TensorFlow 或 PyTorch 的 Checkpoint TensorFlow, PyTorch .h5 Keras/TensorFlow 權重 TensorFlow, Keras .tflite TensorFlow Lite 模型 TensorFlow Lite .msgpack Chainer 權重存儲格式 Chainer .npz JAX 或 NumPy 存儲格式 JAX, NumPy .onnx ONNX 格式，方便跨框架使用 ONNX 當下載 Hugging Face 或 GitHu...