Skip to content

M
models

PaddlePaddle / models
大约 1 年 前同步成功

通知 222
Star 6828
Fork 2962
M
models
未验证 提交 023742e3 编写于 作者: Q qingqing01 提交者： GitHub
浏览文件
操作

Fix one gpu training in deeplabv3+ (#1966)

上级 1af11949
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
fluid/PaddleCV/deeplabv3+/train.py
浏览文件 @ 023742e3
...@@ -184,7 +184,7 @@ if args.parallel: ...@@ -184,7 +184,7 @@ if args.parallel:
build_strategy=build_strategy, build_strategy=build_strategy,
exec_strategy=exec_strategy) exec_strategy=exec_strategy)
else: else:
binary = fluid.compiler.CompiledProgram(main) binary = fluid.compiler.CompiledProgram(tp)
if args.use_py_reader: if args.use_py_reader:
assert(batch_size % fluid.core.get_cuda_device_count() == 0) assert(batch_size % fluid.core.get_cuda_device_count() == 0)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或