Question about memory usage during model training

[LeidiGa]

Hi there,

Firstly, I want to say thank you for the helpful responses to my previous issue (#10 ). I was able to resolve the issue, but now I am encountering a new problem during training. I get the following error: "cuda runtime error (700) : an illegal memory access was encountered at src/roi_align_kernel.cu:292".

I am wondering if this error is related to my GPU memory capacity, and if so, how much memory is required for training the model?

Thank you in advance for any help or advice you can offer!