pytorch-center-loss：Pytorch实现中心损失

【文件属性】：
文件名称：pytorch-center-loss：Pytorch实现中心损失
文件大小：5.65MB
文件格式：ZIP
更新时间：2021-02-05 00:22:55
python computer-vision deep-learning pytorch feature-learning 火炬中心损失 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

【文件预览】：
pytorch-center-loss-master
----.gitignore(1KB)
----transforms.py(200B)
----datasets.py(1KB)
----LICENSE(1KB)
----models.py(2KB)
----center_loss.py(2KB)
----utils.py(2KB)
----README.md(5KB)
----main.py(8KB)
----gifs()
--------center_test.gif(1.09MB)
--------softmax_test.gif(1.86MB)
--------center_train.gif(1.09MB)
--------softmax_train.gif(1.69MB)