From 3fab86ffa6e137329fb989084721e40f5274be64 Mon Sep 17 00:00:00 2001 From: ShenLiang <1422485404@qq.com> Date: Sat, 4 Jan 2020 13:01:07 +0800 Subject: [PATCH] add package for centerloss (#21806) --- python/paddle/fluid/layers/loss.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/loss.py b/python/paddle/fluid/layers/loss.py index 9aaa6afd7e..c0d158f7a7 100644 --- a/python/paddle/fluid/layers/loss.py +++ b/python/paddle/fluid/layers/loss.py @@ -23,7 +23,8 @@ from ..framework import Variable, in_dygraph_mode from .. import core from ..data_feeder import check_type_and_dtype from ..param_attr import ParamAttr -from ..initializer import NumpyArrayInitializer +from ..initializer import NumpyArrayInitializer, Constant +from .. import core __all__ = [ 'center_loss', -- GitLab