提交 b8ac7e07 编写于 作者: G ghostplant 提交者: François Chollet

Fix duplicated argname: num_gpus === parts (#10228)

Signed-off-by: NCUI Wei <ghostplant@qq.com>
上级 4f1ea9af
......@@ -184,7 +184,7 @@ def multi_gpu_model(model, gpus=None, cpu_merge=True, cpu_relocation=False):
batch_size = shape[:1]
input_shape = shape[1:]
step = batch_size // parts
if i == num_gpus - 1:
if i == parts - 1:
size = batch_size - step * i
else:
size = step
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册