未验证 提交 1bee7954 编写于 作者: X Xiaoyao Xi 提交者: GitHub

Update README.md

上级 1acb2cec
......@@ -830,7 +830,7 @@ learning_rate(REQUIRED): str类型。训练阶段的学习率。
batch_size(REQUIRED): int类型。批大小,即每个训练或推理step所使用样本数。
epoch(REQUIRED): int类型。主任务的训练epoch数。
use_cpu (OPTIONAL): bool类型。默认为False。框架默认使用GPU进行单机单卡或分布式训练,若希望使用cpu训练或推理,可将该标志位置True。
use_gpu (OPTIONAL): bool类型。默认为True。框架默认使用GPU进行单机单卡或分布式训练,若希望使用cpu训练或推理,可将该标志位置为False。
warmup_proportion (OPTIONAL): float类型。默认为0。对预训练模型finetuning时的warmup的训练step占预估的全部训练步数的比例。
use_ema (OPTIONAL): bool类型。默认为False。是否开启[ema](https://en.wikipedia.org/wiki/Moving_average#Exponential_moving_average) 进行训练和推理。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册