Skip to content

M
models

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

通知 232
Star 6828
Fork 2962
M
models
提交 a6437859 编写于 作者: L liuwei1031
浏览文件
操作

mark loss as persistable

上级 fb5fa1af
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
PaddleCV/object_detection/train.py
浏览文件 @ a6437859
...@@ -179,6 +179,7 @@ def train(args, ...@@ -179,6 +179,7 @@ def train(args,
predicate=if_exist) predicate=if_exist)
if parallel: if parallel:
loss.persistable = True
build_strategy = fluid.BuildStrategy() build_strategy = fluid.BuildStrategy()
build_strategy.enable_inplace = True build_strategy.enable_inplace = True
build_strategy.memory_optimize = True build_strategy.memory_optimize = True
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或