上传者: 42099936
|
上传时间: 2021-11-10 16:21:10
|
文件大小: 5.65MB
|
文件类型: -
火炬中心损失
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