XueWei


@triple-Mu yes, I can get I can distinguish by batchid. but it's not convenient，is there any way make the output batch size = input batch size？

Thanks a lot, It doesn't seem easy， by the way ，why export torchscript not with nms ?