From f06c74737c6bc7517450cc8d101f7162d9475c63 Mon Sep 17 00:00:00 2001 From: FlyingQianMM <245467267@qq.com> Date: Thu, 18 Jun 2020 10:20:09 +0800 Subject: [PATCH] fix dtype error in retinanet_target_assgin example codes. test=develop (#2222) --- doc/fluid/api_cn/layers_cn/retinanet_target_assign_cn.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/fluid/api_cn/layers_cn/retinanet_target_assign_cn.rst b/doc/fluid/api_cn/layers_cn/retinanet_target_assign_cn.rst index 895433d93..cd37a297b 100644 --- a/doc/fluid/api_cn/layers_cn/retinanet_target_assign_cn.rst +++ b/doc/fluid/api_cn/layers_cn/retinanet_target_assign_cn.rst @@ -68,9 +68,9 @@ retinanet_target_assign gt_boxes = fluid.data(name='gt_boxes', shape=[10, 4], dtype='float32') gt_labels = fluid.data(name='gt_labels', shape=[10, 1], - dtype='float32') + dtype='int32') is_crowd = fluid.data(name='is_crowd', shape=[1], - dtype='float32') + dtype='int32') im_info = fluid.data(name='im_info', shape=[1, 3], dtype='float32') score_pred, loc_pred, score_target, loc_target, bbox_inside_weight, fg_num = \ -- GitLab