@@ -12,18 +12,18 @@ Model
1212参数
1313:::::::::
1414
15- - **network ** (paddle.nn.Layer): 是 ``paddle.nn.Layer `` 的一个实例
16- - **inputs ** (InputSpec|list|dict|None, 可选): ``network `` 的输入,可以是 ``InputSpec `` 的实例,或者是一个 ``InputSpec `` 的 ``list `` , 或者是格式为 ``{name: InputSpec} `` 的 ``dict `` ,或者为 ``None `` . 默认值为 ``None `` .
17- - **labels ** (InputSpec|list|None, 可选): ``network `` 的标签,可以是 ``InputSpec `` 的实例,或者是一个 ``InputSpec `` 的 ``list `` , 或者为 ``None ``. 默认值为 ``None `` .
15+ - **network ** (paddle.nn.Layer): 是 ``paddle.nn.Layer `` 的一个实例
16+ - **inputs ** (InputSpec|list|dict|None, 可选): ``network `` 的输入,可以是 ``InputSpec `` 的实例,或者是一个 ``InputSpec `` 的 ``list ``, 或者是格式为 ``{name: InputSpec} `` 的 ``dict ``,或者为 ``None ``。 默认值为 ``None ``。
17+ - **labels ** (InputSpec|list|None, 可选): ``network `` 的标签,可以是 ``InputSpec `` 的实例,或者是一个 ``InputSpec `` 的 ``list ``, 或者为 ``None ``。 默认值为 ``None ``。
1818
1919.. note ::
2020
21- 在动态图中, ``inputs `` 和 ``labels `` 都可以设置为 ``None ``. 但是,在静态图中, ``input `` 不能设置为 ``None `` . 而如果损失函数需要标签(label)作为输入,则必须设置 ``labels `` ,否则,可以为 ``None `` 。
21+ 在动态图中,``inputs `` 和 ``labels `` 都可以设置为 ``None ``。 但是,在静态图中,``input `` 不能设置为 ``None ``。 而如果损失函数需要标签(label)作为输入,则必须设置 ``labels ``,否则,可以为 ``None `` 。
2222
2323代码示例
2424:::::::::
2525
26- 1.一般示例
26+ 1. 一般示例
2727
2828.. code-block :: python
2929
5959 model.fit(data, epochs = 2 , batch_size = 32 , verbose = 1 )
6060
6161
62- 2.使用混合精度训练的例子
62+ 2. 使用混合精度训练的例子
6363
6464.. code-block :: python
6565
@@ -102,7 +102,7 @@ train_batch(inputs, labels=None)
102102**参数 **
103103
104104 - **inputs ** (list) - 1维列表,每个元素都是一批次的输入数据,数据类型为 ``numpy.ndarray ``。
105- - **labels ** (list) - 1维列表,每个元素都是一批次的输入标签,数据类型为 ``numpy.ndarray `` 。默认值:None。
105+ - **labels ** (list) - 1维列表,每个元素都是一批次的输入标签,数据类型为 ``numpy.ndarray ``。默认值:None。
106106
107107**返回 **
108108
@@ -146,7 +146,7 @@ eval_batch(inputs, labels=None)
146146
147147
148148 - **inputs ** (list) - 1维列表,每个元素都是一批次的输入数据,数据类型为 ``numpy.ndarray `` 。
149- - **labels ** (list) - 1维列表,每个元素都是一批次的输入标签,数据类型为 ``numpy.ndarray `` 。默认值:None。
149+ - **labels ** (list) - 1维列表,每个元素都是一批次的输入标签,数据类型为 ``numpy.ndarray ``。默认值:None。
150150
151151**返回 **
152152
@@ -292,7 +292,7 @@ load(path, skip_mismatch=False, reset_optimizer=False)
292292**参数 **
293293
294294
295- - **path ** (str) - 保存参数或优化器信息的文件前缀。格式如 ``path.pdparams `` 或者 ``path.pdopt `` ,后者是非必要的,如果不想恢复优化器信息。
295+ - **path ** (str) - 保存参数或优化器信息的文件前缀。格式如 ``path.pdparams `` 或者 ``path.pdopt ``,后者是非必要的,如果不想恢复优化器信息。
296296 - **skip_mismatch ** (bool) - 是否需要跳过保存的模型文件中形状或名称不匹配的参数,设置为 ``False `` 时,当遇到不匹配的参数会抛出一个错误。默认值:False。
297297 - **reset_optimizer ** (bool) - 设置为 ``True `` 时,会忽略提供的优化器信息文件。否则会载入提供的优化器信息。默认值:False。
298298
@@ -358,7 +358,7 @@ prepare(optimizer=None, loss=None, metrics=None, amp_configs=None)
358358 - **optimizer ** (Optimizer) - 当训练模型的,该参数必须被设定。当评估或测试的时候,该参数可以不设定。默认值:None。
359359 - **loss ** (Loss) - 当训练模型的,该参数必须被设定。默认值:None。
360360 - **metrics ** (Metric|list[Metric]) - 当该参数被设定时,所有给定的评估方法会在训练和测试时被运行,并返回对应的指标。默认值:None。
361- - **amp_configs ** (str|dict|None) - 混合精度训练的配置,通常是个dict,也可以是str。当使用自动混合精度训练或者纯float16训练时,``amp_configs `` 的key ``level `` 需要被设置为O1或者O2,float32训练时则默认为O0。除了 ``level `` ,还可以传入更多的和混合精度API一致的参数,例如:``init_loss_scaling `` 、 ``incr_ratio `` 、 ``decr_ratio `` 、 ``incr_every_n_steps `` 、 ``decr_every_n_nan_or_inf `` 、 ``use_dynamic_loss_scaling `` 、 ``custom_white_list `` 、 ``custom_black_list `` ,在静态图下还支持传入 ``custom_black_varnames `` 和 ``use_fp16_guard `` 。详细使用方法可以参考参考混合精度API的文档 :ref: `auto_cast <cn_api_amp_auto_cast >` 和 :ref: `GradScaler <cn_api_amp_GradScaler >` 。为了方便起见,当不设置其他的配置参数时,也可以直接传入 ``'O1' `` 、``'O2' `` 。在使用float32训练时,该参数可以为None。默认值:None。
361+ - **amp_configs ** (str|dict|None) - 混合精度训练的配置,通常是个dict,也可以是str。当使用自动混合精度训练或者纯float16训练时,``amp_configs `` 的key ``level `` 需要被设置为O1或者O2,float32训练时则默认为O0。除了 ``level ``,还可以传入更多的和混合精度API一致的参数,例如:``init_loss_scaling `` 、 ``incr_ratio `` 、 ``decr_ratio `` 、 ``incr_every_n_steps `` 、 ``decr_every_n_nan_or_inf `` 、 ``use_dynamic_loss_scaling `` 、 ``custom_white_list `` 、 ``custom_black_list ``,在静态图下还支持传入 ``custom_black_varnames `` 和 ``use_fp16_guard ``。详细使用方法可以参考参考混合精度API的文档 :ref: `auto_cast <cn_api_amp_auto_cast >` 和 :ref: `GradScaler <cn_api_amp_GradScaler >`。为了方便起见,当不设置其他的配置参数时,也可以直接传入 ``'O1' `` 、``'O2' ``。在使用float32训练时,该参数可以为None。默认值:None。
362362
363363
364364fit(train_data=None, eval_data=None, batch_size=1, epochs=1, eval_freq=1, log_freq=10, save_dir=None, save_freq=1, verbose=2, drop_last=False, shuffle=True, num_workers=0, callbacks=None)
427427 save_dir = ' mnist_checkpoint' )
428428
429429
430- 2. 使用Dataloader训练的例子.
430+ 2. 使用Dataloader训练的例子。
431431
432432 .. code-block :: python
433433
0 commit comments