文件名称：sam:SAM
文件大小：619KB
文件格式：ZIP
更新时间：2024-04-09 18:51:36
sam optimizer pytorch sharpness-aware Python
SAM优化器 锐度感知最小化可有效提高泛化能力 〜在Pytorch中〜 SAM同时将损耗值和损耗锐度最小化。特别地，它寻找位于具有均匀低损耗的邻域中的参数。 SAM改进了模型的通用性，并。此外，它提供了强大的鲁棒性，可与专门针对带有噪声标签的学习的SoTA程序所提供的噪声相提并论。 这是的非官方存储库，。在实现方面，SAM类是一个轻量级包装器，用于计算正则化的“清晰度感知”渐变，该渐变由基础优化器（例如带有动量的SGD）使用。该存储库还包括一个的简单 ;作为概念验证，它在此数据集上以强劲的势头击败了SGD的表现。 在使用和不使用SAM的情况下，培训结束时的ResNet损失情况。锐度感知更新导致最小值明显变宽，从而导致更好的泛化属性。 用法 在您的培训管道中使用SAM应该很简单。只需记住，训练的速度将慢一倍，因为SAM需要两次向前-向后的传递才能使“清晰感”梯度变准。如果您使用渐变剪切，请
【文件预览】：
sam-main
----.gitignore(2KB)
----LICENSE(1KB)
----.github()
--------stale.yml(682B)
----sam.py(2KB)
----README.md(5KB)
----example()
--------data()
--------model()
--------utility()
--------README.md(172B)
--------train.py(3KB)
----img()
--------loss_landscape.png(630KB)
--------README.md(80B)