To be honest I checked it on task manager. On linux I get runtime error during training (I suspected that it was related with small ram capacity of my gpu, but when I use CPU my RAM fills up during  a training )

image.png871×164 3.94 KB

Code :

image.png555×572 19.1 KB