Solved by spanev in post #9 
              
              
                Can you share the code of the model or at least the concerned part of the model? 
If out is nested in another layer or class you would have to do something like model.submodule_name.out