search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

How can I do inference without mynn.DataParallel? #164

Open shiyi001 opened 5 years ago

shiyi001 commented 5 years ago

I used mynn.DataParallel during inference but found it costs lots of cpu.

How can I do inference without mynn.DataParallel?

achalddave commented 5 years ago

Do you get an error if you just remove the line that wraps the model in mynn.DataParallel?

shiyi001 commented 5 years ago

I am sorry to bother you.

I really met some error when I removed mynn.DataParallel, but those were easy to fix.

Please close this issue. Thanks for your reply.

1179021477 commented 4 years ago

Excuse me, can you tell me what problem you have when removing mynn.DataParallel, and how to fix it?

shiyi001 commented 4 years ago

Excuse me, can you tell me what problem you have when removing mynn.DataParallel, and how to fix it?

Sorry for the long time passed, I have forgotten the problem I encountered. But I remember that only a few changes are needed to fix this problem.