PyTorch中文文档细读
文章平均质量分 79
PyTorch中文文档细读
UvwxyZ666
这个作者很懒,什么都没留下…
展开
-
笔记:PyTorch文档阅读(3)torchvision.models
以下是pytorch中文文档原文,根据自己的需要,可以选择合适的模型直接下载预训练好的模型,再调整最后一节全连接层输出。这样可以快速收敛,并且不用自己搭建网络。原创 2024-08-09 08:23:17 · 1044 阅读 · 0 评论 -
笔记:PyTorch文档阅读(2)序列化语义即保存模型
第一种推荐的方法,即只保存和加载模型参数,只涉及到模型的权重和偏置等参数,而不包括模型的架构。在加载模型时,需要先实例化模型架构,然后再将保存的参数加载到这个架构中。指的是将数据结构或对象的状态转换为可以存储或传输的格式的过程,以及将这些数据重新转换回原始状态的过程。然而,在这种情况下,序列化的数据被绑定到特定的类和固定的目录结构,所以当在其他项目中使用时,或者在一些严重的重构器之后它可能会以各种方式break。这样,模型的架构和参数就都恢复了,模型可以用于进一步的推理或继续训练。将它们保存到文件中。原创 2024-08-05 08:42:33 · 371 阅读 · 0 评论 -
笔记:PyTorch文档阅读(1)自动求导机制
如果你需要实现自定义的梯度计算或者使用梯度裁剪等技术,梯度跟踪是必要的。原创 2024-08-04 09:45:28 · 896 阅读 · 0 评论