Update center_loss.py - Githubissues

KaiyangZhou / pytorch-center-loss

Pytorch implementation of Center Loss

MIT License

965 stars 218 forks source link

Update center_loss.py #6

Closed Chuhanxx closed 5 years ago

Chuhanxx commented 5 years ago

Improved the efficiency of calculating distance to feature centres by using a matrix multiplication instead of for loop, decreasing the computation time to be 1/3 of the original one.