DeepSpeed
28b9d5c2 - Add condition when dimension is greater than 2 (#4390)

Commit

2 years ago

Add condition when dimension is greater than 2 (#4390) When dimension is greater than 2, grad_bias should be calculated differently.

References

Author

ajindal1

Parents