roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I do inference without mynn.DataParallel?

shiyi001 opened this issue · comments

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

How can I do inference without mynn.DataParallel?

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

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.

commented

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

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.