Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
No need to call
xm.mark_step()
explicitly (#4)
Since for gradient accumulation we're accumulating on batches from `ParallelLoader` instance which on next() marks the step itself.
- Loading branch information