python如何训练和保存模型

ZEERO~

已于 2024-01-07 21:14:59 修改

阅读量394

点赞数

分类专栏：机器学习深度学习 PYTHON 文章标签：机器学习

于 2023-10-17 00:45:03 首次发布

本文链接：https://blog.csdn.net/weixin_43249038/article/details/133874814

版权

PYTHON 同时被 3 个专栏收录

19 篇文章 3 订阅

订阅专栏

机器学习

11 篇文章 0 订阅

订阅专栏

深度学习

10 篇文章 0 订阅

订阅专栏

1、ANN模型

import torch
import torch.nn as nn

# 创建一个示例模型
model = nn.Sequential(
    nn.Linear(10, 5),
    nn.ReLU(),
    nn.Linear(5, 2)
)

# 保存模型到文件
torch.save(model.state_dict(), 'model.pth')

import torch
import torch.nn as nn

# 创建一个与之前示例模型结构相同的实例
model = nn.Sequential(
    nn.Linear(10, 5),
    nn.ReLU(),
    nn.Linear(5, 2)
)

# 加载保存的模型参数
model.load_state_dict(torch.load('model.pth'))