Skip to content

Commit 5357e6f

Browse files
committed
docs(solver): 求解器实现
1 parent 7a13f66 commit 5357e6f

File tree

2 files changed

+34
-1
lines changed

2 files changed

+34
-1
lines changed

docs/求解器.md

+32
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
2+
# Sovler
3+
4+
模型的整个训练在`solver.py`文件中实现
5+
6+
```
7+
class Solver(object):
8+
9+
def __init__(self, model, data, criterion, optimizer, **kwargs):
10+
11+
def _reset(self):
12+
13+
def _step(self, X_batch, y_batch):
14+
15+
def check_accuracy(self, X, y, num_samples=None, batch_size=8):
16+
17+
def train(self):
18+
```
19+
20+
## __init__
21+
22+
* 必选参数
23+
* `model`:网络模型
24+
* `data`:包含了训练和测试数据集
25+
* `criterion`:评价函数
26+
* `optimizer`:优化器
27+
* 可选参数
28+
* `lr_scheduler`:学习率调度器,默认为`None`
29+
* `batch_size`:单次处理大小,默认为`8`
30+
* `num_epochs`:迭代周期次数,默认为`10`
31+
* `reg`:正则化因子,默认为`1e-3`
32+
* `print_every`:每隔多少论打印一次信息,默认为`1`

mkdocs.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -32,4 +32,5 @@ nav:
3232
- Home: index.md
3333
- 整体架构: overall-structure.md
3434
- 网络层: 网络层.md
35-
- 网络模型: 网络模型.md
35+
- 网络模型: 网络模型.md
36+
- 求解器: 求解器.md

0 commit comments

Comments
 (0)