Hi ptrblck,
Thank you very much! I try your command and get the following information:

image.png800×453 13.4 KB


image.png1415×828 41.6 KB

I notice that before the epoch to tell me out of memory, as shown in the blue line, GeForce GTX card seems to be almost doubled suddenly and it fails:

image.png810×602 19.6 KB

Is there any way in pytorch to use the another GPU card in sequence to satisfy it?
Or should I need to reduces number CONV layer or number of channels in my model?
Thank you very much!
Best,
Peter