I have the same issue. Did you figure out the problem? Wanted to know if anyone had similar issue and if it is fixed, I am not looking for answers Would appreciate your feedback.
Hello @Balaji_K , I had the same issue and I even replied to your message. But I found the issue and fixed it. I hope you can solve the issue as well soon. All the best.
I would suggest going through your function definition once more and I am sure you will find it as well (if we both made same error)
The most common mistake is using global variable word_embeddings instead of the local embeddings variable that you should use.
So please check if you are not using word_embeddings inside the function.