accelerate
79ca85c2 - Support skip_first_batches for XLA (#2966)

Commit

1 year ago

Support skip_first_batches for XLA (#2966) * Fix skip_first_batches for XLA * Use state to check XLA * Change to PartialState

References

Author

yitongh

Parents