1 Like
I have same problem with that, need help
1 Like
Try to reduce the batch_size. It’s work for me
2 Likes
Thanks, it solved for me too
1 Like
glad to hear my solution helped you
1 Like
how much for the batch_size?
2 Likes
In my lab I set to 32
1 Like
I have the same problem and it still dies at 32.
Maybe you can try reduce batch_size again