Your GPU is running out of memory, so you would need to reduce the memory usage e.g. via reducing the batch size.



 Najeh_Nafti:

(Not related to the memory)


I’m not sure how to understand this information, as the error points towards the OOM issue.