No, I want to do model parallel. In the above snippet I used a for loop, so I think that it is not parallel processing.
I noticed that nn.parallel.parallel_apply(modulelist, inputs) is suitable, is this correct?
When actually comparing the speed of for loop and parallel_apply, I feel that parallel_apply is a little faster.