model.detect_objects take so much time

sgrvinod / a-PyTorch-Tutorial-to-Object-Detection

SSD: Single Shot MultiBox Detector | a PyTorch Tutorial to Object Detection

MIT License

3.04k stars 718 forks source link

model.detect_objects take so much time #11

Closed lvmy3 closed 5 years ago

lvmy3 commented 5 years ago

when I try the code, the model.detect_objects part take much more time than training part, is that normal? I am using 4gpu, but model.detect_objects actually use only one(model.module.detect_objects), so it's very slow. Is there any possible way to improve this?

gavrilo91 commented 4 years ago

@lvmy3 Did you solved this issue? Thanks :)

shenoynikhil98 commented 3 years ago

Any solutions here?