lucasnfe / bardo-composer Goto Github PK

Hi,
we are trying to reproduce the results of the music classifier part for academic reasons,
we faced a problem at the beginning because we had to run the midi_encoder.py file first to create the vocab and to create the text files for the midi files , it's not mentioned in the steps though.
now after this step we are trying to start with the next script to finetune the gpt-2 but we face the next error and we thought you could help us with it
each time we try to run this script

it starts training and in the first epoch 1/10 the error occur at different iteration each run and it's as follows

NOTE : we are running on Colab
Thanks in advance,