Hi,
The error is that during the backward pass, when it tries to allocate memory to store the gradients and perform computations, there is not enough. You should try reducing the batch size.
2 Likes
Hi,
The error is that during the backward pass, when it tries to allocate memory to store the gradients and perform computations, there is not enough. You should try reducing the batch size.