I am working on  a network ,where ,I loaded a part of network parameters by  using   .children  and then accessing it by modules in forward . The code working fine , but only with one GPU card even though nn.parallel (net) was given. To my knowledge problem is while giving input to network. Instead of giving net(input) during training ,I m assigning net.modules(input),because an error is popping while using net(input).
Can, you please suggest me ways to correct my error