Thanks! I will do that . I am also running into another issue “IndexError: Target 20 is out of bounds.” when it starts to train. Can you please tell me what’s causing this error? Thanks again in advance!