Do you mean ram memory or GPU memory?
I’m actually encountering a Segmentation Fault (core dumped). My fear is that It’s related to this problem https://github.com/pytorch/pytorch/issues/31758 7.
I’m running the script in debug, I’ll update the post as soon as i get the error.
Also, I got the error just by running an empty training loop, just loading the data into RAM.