Solved by chari8 in post #3 
              
              
                Hi, ptrblck! Thank you for the reply. 
First, to answer your question, the original images were resized to 152 * 242 pix, and I used batch_size = 1, but still got memory shortage problem before the first epoch finishes. 
Here is a good news (well good for me). I solved the problem by substitute all …