re3-pytorch icon indicating copy to clipboard operation
re3-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon danielgordon10

Code very slow

Open LilianLaporte opened this issue 1 year ago • 1 comments

Hello,

When I run the image_demo.py, the creation of the instance tracker and of the first call of track take a lot of time (several minutes). If I am using the small weights it does not change the problem. On another hand, the version with tensorflow is really fast and works properly. Is it normal if it is that slow or should it be fast?

NB: my GPU is a NVIDIA A4000

LilianLaporte avatar Dec 20 '23 11:12 LilianLaporte

No, it should be fast. Can you add some timing statements to figure out which line is causing the slowdown? Maybe torch is trying to set up some optimization under the hood.

danielgordon10 avatar Jan 03 '24 23:01 danielgordon10