Skip to content

Commit 23a96bd

Browse files
📝 更新文档,优化PyTorch和OpenCV相关内容
- 删除了不再需要的产品评测文档,简化文档结构。 - 在PyTorch部分新增了模型保存与加载的示例,提升了实用性。 - 更新了Transformer结构的标题和位置,确保信息的清晰性。 - 增加了OpenCV常用算法的详细说明,丰富了图像处理相关内容。
1 parent 9563880 commit 23a96bd

File tree

9 files changed

+659
-69
lines changed

9 files changed

+659
-69
lines changed

docs/docs/大模型的应用/产品评测.md

Lines changed: 0 additions & 5 deletions
This file was deleted.

docs/docs/机器学习/Transformer/Transformer结构.md

Lines changed: 0 additions & 9 deletions
This file was deleted.

docs/docs/机器学习/Transformer/index.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
---
2-
sidebar_position: 8
3-
title: Transformer
2+
sidebar_position: 7
3+
title: Transformer结构
44
---
55

66
学习Transformer的原理,使用PyTorch实现一个简单的Transformer模型,对模型进行训练和推理。通过分布式训练,提升模型训练效率。
@@ -57,4 +57,4 @@ $$
5757

5858

5959

60-
<DocCardList />
60+

docs/docs/机器学习/index.md

Lines changed: 133 additions & 31 deletions
Large diffs are not rendered by default.

docs/docs/机器学习/决策树.md

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,53 @@ os.remove('new_app_model_v1.pkl')
206206
```
207207

208208

209+
### torch模型的保存与加载
210+
211+
可选择保存整个模型或仅参数,原生支持分布式/大模型。
212+
213+
保存函数签名:`torch.save()`
214+
215+
加载函数签名:`torch.load()`
216+
217+
示例代码:
218+
219+
```python
220+
import torch
221+
import torch.nn as nn
222+
223+
# 定义一个简单的模型
224+
class SimpleNet(nn.Module):
225+
def __init__(self, input_size, hidden_size, output_size):
226+
super(SimpleNet, self).__init__()
227+
self.fc1 = nn.Linear(input_size, hidden_size)
228+
self.relu = nn.ReLU()
229+
self.fc2 = nn.Linear(hidden_size, output_size)
230+
231+
def forward(self, x):
232+
x = self.fc1(x)
233+
x = self.relu(x)
234+
x = self.fc2(x)
235+
return x
236+
237+
# 创建模型实例
238+
model = SimpleNet(input_size=784, hidden_size=128, output_size=10)
239+
240+
# 方法1:保存整个模型(包括结构和参数)
241+
torch.save(model, 'complete_model.pth')
242+
243+
# 方法2:只保存模型参数(推荐)
244+
torch.save(model.state_dict(), 'model_weights.pth')
245+
246+
# 方法1对应的加载:加载整个模型
247+
loaded_model = torch.load('complete_model.pth')
248+
249+
# 方法2对应的加载:加载模型参数
250+
model_for_loading = SimpleNet(input_size=784, hidden_size=128, output_size=10)
251+
model_for_loading.load_state_dict(torch.load('model_weights.pth'))
252+
253+
```
254+
255+
209256
## 随机森林
210257

211258
随机森林是对决策树集合的特有名称。

docs/docs/机器学习/自然语言处理/DNN.md

Whitespace-only changes.

docs/docs/机器学习/自然语言处理/RNN.md

Whitespace-only changes.

docs/docs/机器学习/自然语言处理/index.md

Lines changed: 0 additions & 7 deletions
This file was deleted.

0 commit comments

Comments
 (0)