The REID part costs time too much

[HavenLan]

Hello, we run the code successfully, but we find a question of the time costing. In the sort part, the average time is about 40ms, but there are few frames taking more than 500ms. After investigation, it is determined that the time-consuming part of network (feature extraction part) reasoning has increased sharply, and the Layer4 takes the most time. Have you encountered any similar problems? How to solve them, thanks!

Environment: Xavier AGX 16GB - Jetpack 4.4 python 3.6.9 pytorch 1.6.0 opencv 4.1.1