Ryo's Blog
  • Articles
  • Learn from AI
  • Tags
  • More
  • ...
#杂技#Blog#杂谈#C++#Python#数据结构#算法#排序#算法竞赛#设计模式#笔记#GitHub#AWS#CI/CD#IaC#DevOps#VSCode#Hexo#JavaScript#Kubernetes#Docker#Cloud Native#Cursor#可视化#数学#工具#Nextjs#Tag1#Tag2#单行Tag#Tag#Tag/Tag2#Tag3#Linux#systemctl#journalctl#timedatectl#BasicKnowledge#Operation#Signal#memory#schedule#Cloud Computing#PyTorch#ONNX#深度学习#模型部署#学习笔记#OpenCV#图像处理#Rust#编程语言#Windows#macOS#可执行文件#LoRA#参数高效微调#模型训练#反向传播#自动微分#GoLang#Memory Management#Deep Learning#Model Management#Neural Networks#ResNet#Mathematics#协议#IoT#物联网#消息队列#Stable Diffusion#Concurrency#Scheduler#Goroutine#test#wiki#page1#subpage1#page2#subpage2#subpage3#subpage34#page3
深度学习模型文件格式对比:pth、safetensor 与 onnx
2024-11-29

> 本页面介绍了三种机器学习模型文件格式:.pth、.safetensor 和 .onnx。主要内容包括每种格式的定义、保存的内容、用途以及它们之间的区别和转换方法。

PyTorch 中的`.pth`文件通常用于保存模型的权重(parameters)和训练过程中的优化器状态(optimizer state)。具体来说,一个`.pth`文件可能包含以下内容:

1. **模型状态字典(Model state dictionary)**:

#PyTorch#ONNX#深度学习#模型部署#学习笔记
© 2023 Ryo Jerry Yu. All rights reserved.