[Bug]fix bug about softmax

Angel-ML / PyTorch-On-Angel

PyTorch On Angel, arming PyTorch with a powerful Parameter Server, which enable PyTorch to train very big models.

165 stars 51 forks source link

Closed rachelsunrh closed 3 years ago

rachelsunrh commented 3 years ago

fix bug about softmax