pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取

pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第1张图片
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第2张图片
false:网络架构替换,参数默认(没训练)
true:(训练好的)

pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第3张图片
如何利用现有网络进行修改:
vgg分为100类–》10类:
(1)加入
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第4张图片
层加到classifier中:
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第5张图片
(2)修改
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第6张图片

2.保存

(1)
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第7张图片
在这里插入图片描述
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第8张图片
(2)第二种方式更小
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第9张图片
恢复成网络模型
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第10张图片
(3)陷阱(方式1)
保存:
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第11张图片
加载:
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第12张图片
应该确保网络模型是想要的模型(要重新引入class模型)—》其实是可以单独引入到model.load中(import)
pytorch深度学习基础快速入门6:现有网络模型的使用及修改、网络模型的保存与读取_第13张图片

你可能感兴趣的:(python,深度学习,pytorch,人工智能)