未验证 提交 ed4bbe08 编写于 作者: S Stas Bekman 提交者: GitHub

[config] fix assert message (#1734)

上级 9351266f
......@@ -995,8 +995,8 @@ class DeepSpeedConfig(object):
), f"Gradient accumulation steps: {grad_acc} has to be greater than 0"
assert train_batch == micro_batch * grad_acc * self.world_size, (
f"Check batch related parameters. train_batch_size is not equal"
" to micro_batch_per_gpu * gradient_acc_step * world_size"
f"Check batch related parameters. train_batch_size is not equal "
"to micro_batch_per_gpu * gradient_acc_step * world_size "
f"{train_batch} != {micro_batch} * {grad_acc} * {self.world_size}"
)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册