Apex Batch Size In Neural Network