From acfcb1c5e1db40a9b6da7fa7e5ff4daea98780f0 Mon Sep 17 00:00:00 2001 From: wangxiao Date: Wed, 4 Dec 2019 11:56:28 +0800 Subject: [PATCH] change to fluid.one_hot --- paddlepalm/mtl_controller.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlepalm/mtl_controller.py b/paddlepalm/mtl_controller.py index 1d229c7..53c57a0 100755 --- a/paddlepalm/mtl_controller.py +++ b/paddlepalm/mtl_controller.py @@ -453,7 +453,7 @@ class Controller(object): # compute loss task_id_var = net_inputs['__task_id'] - task_id_vec = layers.one_hot(task_id_var, num_instances) + task_id_vec = fluid.one_hot(task_id_var, num_instances) losses = fluid.layers.concat([task_output_vars[inst.name+'/loss'] for inst in instances], axis=0) loss = layers.reduce_sum(task_id_vec * losses) -- GitLab