fix infer error.

revert-16190-refine_parallel_executor
dengkaipeng 7 years ago
parent c9d4676bee
commit deec3ac1e9

@ -655,8 +655,6 @@ def yolo_box(x,
raise TypeError("Input img_size of yolo_box must be Variable")
if not isinstance(anchors, list) and not isinstance(anchors, tuple):
raise TypeError("Attr anchors of yolo_box must be list or tuple")
if not isinstance(anchor_mask, list) and not isinstance(anchor_mask, tuple):
raise TypeError("Attr anchor_mask of yolo_box must be list or tuple")
if not isinstance(class_num, int):
raise TypeError("Attr class_num of yolo_box must be an integer")
if not isinstance(conf_thresh, float):
@ -668,7 +666,7 @@ def yolo_box(x,
attrs = {
"anchors": anchors,
"class_num": class_num,
"conf_thresh": ignore_thresh,
"conf_thresh": conf_thresh,
"downsample_ratio": downsample_ratio,
}

Loading…
Cancel
Save