diff --git a/PaddleCV/object_detection/train.py b/PaddleCV/object_detection/train.py index 0b660a94f201ce05a0273e8c0bf7d4c727620944..ebf969ef0655557feca405c3eb0e558c9588276d 100644 --- 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