training checkpoints don't work

[zoineb]

dear all
thank you for the open source training code and i am impressed by the pesto algorithm performance, you did a great job.

i was trying to train pesto model on mir1k dataset and i managed to start the training and everything worked amazing but when i use the model's output checkpoints from the training i receive a straight constant value graph, if you have any idea on why this may be happening with me please let me know

thank you so much