Google Colab after running this TensorFlow code: “Your session crashed for an unknown reason.” “an illegal memory access was encountered”

Question

I\'m coding a simple TensorFlow code on Colab. I\'m trying to code Word2Vec from scratch (using the method here). neighbors here are just a bunch of tuples like