bugfix

src/infer/trt/yolo/yolo.cu

@@ -536,8 +536,8 @@ std::shared_ptr<data::InstanceSegmentMap> YoloModelImpl::decode_segment(int imem
         checkRuntime(cudaMemcpyAsync(original_mask_out_host, original_mask_out_device,
                                     original_mask_out.gpu_bytes(),
                                     cudaMemcpyDeviceToHost, stream_));
-        return seg;
     }
+    return seg;
     
 }