Solved by karmus89 in post #5 
              
              
                Hi @AlexisW! 
I faced a similar kind of problem. In my case I had to go through my self-implemented initialization function, that essentially assigns random values to several hyperparameters. The key culprit was in value types. I had to transform all instances of numpy variables to original Python t…