- Add current batch record in training manager
- Move zero gradients to `optimize` function in training manager