机器学习-网络模型的保存于读取(pytorch环境)
2021/8/28 23:06:48
本文主要是介绍机器学习-网络模型的保存于读取(pytorch环境),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
例子
import torchvision from torch import nn vgg16_false = torchvision.models.vgg16(pretrained=False) vgg16_true = torchvision.models.vgg16(pretrained=True)
vgg16_true.classifier.add_module('add_liner',nn.Linear(1000,10))的作用:在classifier的Sequential中添加一个名为‘add_linear'的层
vgg16_false.classifier[6] = nn.Linear(4096,10)的作用:将classifier中的Sequential第7个元素修改为nn.Linear(4096,10)
这篇关于机器学习-网络模型的保存于读取(pytorch环境)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-29Elasticsearch慢查询日志配置
- 2024-05-29揭秘华为如此多成功项目的产品关键——Charter模板
- 2024-05-29海外IDC业务拓展的7大挑战
- 2024-05-29InLine Chat功能优化对标Github Copilot,CodeGeeX带来更高效、更直观的编程体验!
- 2024-05-29CodeGeeX 智能编程助手 6 项功能升级,在Visual Studio插件市场霸榜2周!
- 2024-05-29AutoMQ 生态集成 Apache Doris
- 2024-05-292024年IDC行业的深度挖掘:机遇、挑战与未来展望
- 2024-05-29五款扩展组件齐发 —— Volcano、Keda、Crane-scheduler 等,邀你体验
- 2024-05-29AutoMQ 对象存储数据高效组织的秘密: Compaction
- 2024-05-29活动预告|来 GIAC 大会听大数据降本利器:AutoMQ 基于云原生重新设计的 Kafka