Hi @albanD
Thank you so much for your reply!
No I did not share GPU with anyone.
The 11+GB is my local ram but gpu memory. I’m using a 2080 so it has only 8 GB of vram.
The thing is it fails every single time if I set input size = 320, batch = 9. However if I do input size = 320, batch = 16 right after that, it runs perfectly every single time too.
I tried to clear cache before back propagation too but the error incurred .
Thank you again for helping me!