diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py index 4299b17ebd23470fe35d84acb197616a55a15dd0..76722f26007c4aa9d673929e2065ddbfed644256 100644 --- a/python/paddle/nn/functional/loss.py +++ b/python/paddle/nn/functional/loss.py @@ -41,6 +41,7 @@ from ...fluid.layers import teacher_student_sigmoid_loss #DEFINE_ALIAS from ...fluid.layers import edit_distance #DEFINE_ALIAS from ...fluid.layers import sampled_softmax_with_cross_entropy #DEFINE_ALIAS +from ...fluid.layers import huber_loss from ...fluid.layer_helper import LayerHelper from ...fluid.framework import in_dygraph_mode from ...fluid.framework import _varbase_creator