Solved by albanD in post #2 
              
              
                Hi, 
You will need to use nn.ModuleList() for the submodules (each of your regressor) to be found properly. 
You can simply replace self.branches=[] by self.branches=nn.ModuleList() and it will work as expected.