The DeprecationWarning is raised by numpy not PyTorch while the “Out of memory” error is raised by PyTorch if your GPU is running out of memory.
Try to decrease the memory usage by e.g. reducing the batch size.
The DeprecationWarning is raised by numpy not PyTorch while the “Out of memory” error is raised by PyTorch if your GPU is running out of memory.
Try to decrease the memory usage by e.g. reducing the batch size.