I am unable to reproduce the error, with code snippet containing the model and some random input 
I ll keep trying and revert as and when I ll have the error reproduced.
Instead of simplifying various steps I ll keep the code as is