Code terminates without training while using accelerate

[jaydeepb]

I see, thanks! So would it automatically run the model on multiple GPUs?

Also, I was able to figure out the issue. Turns out it was memory error