An error when run "save_logprob"

[mariamartinmarin]

Hello everyone.
First of all, I would like to thank the author for his magnificent work.
Now I expose you my problem.
I have managed to train the model with my own dataset, but then when I try to run "save_logprob" the following error occurs:

File "geotracknet.py", line 157, in <module>
     D["seq"] = np.zeros(tar[:seq_len_d,d_idx_inbatch])[1].reshape(-1,4)
ValueError: cannot reshape array of size 574 into shape(4)

It seems that the array to which we apply the reshape method does not have the necessary elements to be able to group 4 by 4. Does anyone know to what can be due?

I would be very grateful if someone answers me.
Best regards and thanks!