Stage1 training crashes on eval - continued

[kruzel]

hi, thanks for a very interesting paper and supporting code. I'm trying to run training, but fail.

I tried to create the dataset few times, but it didn't help. dataset preparation looks ok, here are the results: