training stuck

[oraby8]

i get an issue during training where the process is getting stuck at the gather stage. specifically, at this progress point:

gather: 75%|█████████████████████████████████████████████████████████████████████████████████████▌ | 9/12 [00:24<00:05, 1.79s/it, reward=0, correct=0, completion_tokens=82]