How can I do inference without mynn.DataParallel?

[shiyi001]

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

How can I do inference without mynn.DataParallel?

[achalddave]

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

[shiyi001]

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]

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

[shiyi001]

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.