Problems: 1. creating mini batches. 2. CUDA memory error

Got it. I think this thread will give you what you need to retrain your model: