Please add force_on_cpu: true in nms_config.
nms_config {
confidence_threshold: 0.25
clustering_iou_threshold: 0.5
force_on_cpu: true
top_k: 200
}
Please add force_on_cpu: true in nms_config.
nms_config {
confidence_threshold: 0.25
clustering_iou_threshold: 0.5
force_on_cpu: true
top_k: 200
}