trying to get minibatches to work