Minor bug. Markdown says batch size is 128 but the actual code has 32.
that earlier mention is not for the cell you are mentioning I guess. Check at the beginning once, what does it mentions there?
