You cannot do that.
The closest thing you can do is to assign data parallel between 2 gpus for module 1 and other 2 for module 2.
If you hard code gpus inside the nn.module you cannot use data parallel