accelerate
Support skip_first_batches for XLA
#2966
Merged

Commits
  • Fix skip_first_batches for XLA
    yitongh committed 2 years ago
  • Use state to check XLA
    yitongh committed 2 years ago
  • Change to PartialState
    yitongh committed 2 years ago
Loading