DeepSpeed
7cab55c7 - Checks for None tensors and skip them when splitting the buckets in zero stage 2. (#728)

Commit

4 years ago

Checks for None tensors and skip them when splitting the buckets in zero stage 2. (#728) * check none tensors when splitting buckets

References

Author

cli99

Parents