火炬中心损失
Pytorch实现中心损失的方法:
( )也使用此损失函数。
开始吧
克隆此仓库并运行代码
$ git clone https://github.com/KaiyangZhou/pytorch-center-loss
$ cd pytorch-center-loss
$ python main.py --eval-freq 1 --gpu 0 --save-dir log/ --plot
您将在终端中看到以下信息
Currently using GPU: 0
Creating dataset: mnist
Creating model: cnn
== > Epoch
1