RBF软件
是一种简单但有效的深度神经网络图像分类损失函数。 用编写并从修改过的RBF-Softmax项目。
在RBF-Softmax中,对数由RBF内核计算,然后由超参数缩放。 因此,此处将最后一个FC中的权重视为类原型。
RBF-Softmax管道
MNIST玩具演示可视化的RBF-Softmax和其他损失。
RBF软件
gif后面是在MNIST上训练的RBF-Softmax的2D特征可视化。 随着训练的进行,内部阶级的距离越来越小。
功能可见。
介绍
培训和测试RBF-Softmax
由于RBF-Softmax使用作为代码库,因此大多数安装遵循pycls。 请参考以获得安装说明。
安装后,请参阅以获取有关使用RBF-Softmax进行培训和评估的基本说明和示例命令。
模型动物园
我们提供了一些最终结果和预训练的模型,可以在下载。 请注意,我们在论文中报告了整个培训期
2021-04-07 10:48:22
4.17MB
Python
1