search

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

fix bug about softmax #64

Closed rachelsunrh closed 3 years ago

rachelsunrh commented 3 years ago

fix bug about softmax method, close #63