Error in baseline.py

[ameet-1997]

There seems to be the following error that I am getting.

tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[0,0] = 3440 is not in [0, 3231)
[[Node: output/embedding_lookup_2 = Gather[Tindices=DT_INT32, Tparams=DT_FLOAT, _class=> > ["loc:@embedding/embedding_W"], validate_indices=true, > _device="/job:localhost/replica:0/task:0/cpu:0"](embedding/embedding_W/read, _arg_input_x_3_0_2)]]

I have tried both Tensorflow version 1.3 with python 2 and Tensorflow 1.6 with python 3 (this one after making the necessary changes).

Any suggestions?