still trying to see why the batch size is zero