From bd8b0ebaba5f13ee31ff7729b7bf6f6728e43001 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Tue, 12 Nov 2019 14:05:00 +0800 Subject: [PATCH] fix distiller typo, test=develop (#21070) --- python/paddle/fluid/contrib/slim/distillation/distiller.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/contrib/slim/distillation/distiller.py b/python/paddle/fluid/contrib/slim/distillation/distiller.py index f08e0bcfefc..98e2326bc56 100644 --- a/python/paddle/fluid/contrib/slim/distillation/distiller.py +++ b/python/paddle/fluid/contrib/slim/distillation/distiller.py @@ -269,7 +269,7 @@ class SoftLabelDistillerPass(object): t_fea = layers.softmax(teacher_feature_map / self.teacher_temperature) t_fea.stop_gradient = True - ce_loss = layres.reduce_mean( + ce_loss = layers.reduce_mean( layers.cross_entropy( s_fea, t_fea, soft_label=True)) distillation_loss = ce_loss * self.distillation_loss_weight -- GitLab