I have uploaded some test data to my github repo 90. If you have time maybe you can try it:

train a model:
python train_attn.py

load the model and do some inferences:
python eval.py


May this is useful to provide some information for solving the problem. Thank you!