Anyway the error suggests that the issue could be resolved by moving NestedTestor to a pytorch class, if it works would you accept a PR on that?

Originally posted by @gillpeacegood in #469 (reply in thread)

Actually there are some historical reasons for the existence of NestedTensor, which is not required since #293 was merged. I guess there's some way to not rely upon NestedTensor for the pre-processing.

https://github.com/zhiqwang/yolov5-rt-stack/blob/1043ee3eacd8218c39e324c39693e373daf29a09/yolort/models/transform.py#L13-L20