Hi, I tried your model with customized datasets using chloroplast genomes in exactly the same format documented, but by checking the log information, here is what I found:

When picking curriculum to True, the epoch 2 and epoch 3 were not entered.
I also inspected the training code, quite strange on the missing actions.