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.