change to fluid.one_hot

acfcb1c5 · wangxiao · ec741693 · acfcb1c5
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddlepalm/mtl_controller.py paddlepalm/mtl_controller.py +1 -1

未找到文件。
--- 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)