search

KaiyangZhou / pytorch-center-loss

Pytorch implementation of Center Loss
MIT License
965 stars 218 forks source link

UserWarning: This overload of addmm_ is deprecated #16

Open phillity opened 4 years ago

phillity commented 4 years ago

Packages: torch 1.5.0 torchvision 0.6.0

Code: distmat.addmm_(1, -2, x, self.centers.t())

Warning: ..\torch\csrc\utils\python_argparser.cpp:756: UserWarning: This overload of addmm is deprecated:      addmm(Number beta, Number alpha, Tensor mat1, Tensor mat2) Consider using one of the following signatures instead:      addmm(Tensor mat1, Tensor mat2, *, Number beta, Number alpha)

Fix: distmat.addmm_(x, self.centers.t(), beta=1, alpha=-2)