From 85ff9692ac616569e39e7f8e0fa6ecca87db5a0b Mon Sep 17 00:00:00 2001 From: yangfukui Date: Tue, 19 Nov 2019 20:28:03 +0800 Subject: [PATCH] Update single_distiller.py --- paddleslim/dist/single_distiller.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleslim/dist/single_distiller.py b/paddleslim/dist/single_distiller.py index 88d34d3b..b6f9a45e 100644 --- a/paddleslim/dist/single_distiller.py +++ b/paddleslim/dist/single_distiller.py @@ -162,7 +162,7 @@ def soft_label_loss(teacher_var_name, return soft_label_loss -def self_defined_loss(program, loss_func, **kwargs): +def loss(program, loss_func, **kwargs): """ Combine variables from student model and teacher model by self defined loss. Args: -- GitLab