mark loss as persistable

a6437859 · liuwei1031 · fb5fa1af · a6437859
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

PaddleCV/object_detection/train.py PaddleCV/object_detection/train.py +1 -0

未找到文件。
--- a/PaddleCV/object_detection/train.py
+++ b/PaddleCV/object_detection/train.py
@@ -179,6 +179,7 @@ def train(args,
                           predicate=if_exist)

    if parallel:
+        loss.persistable = True
        build_strategy = fluid.BuildStrategy()
        build_strategy.enable_inplace = True
        build_strategy.memory_optimize = True