GitHub
ストックにはログインが必要です
PyTorch向けの微分可能な sign / round / floor 関数
Artificial Intelligence
Developer Tools
GitHub
Open Source
概要
- PyTorch 向けの微分可能な sign/round/floor 関数を提供します。
- 決定境界の ε バンド内だけを線形化し、他はハードのまま維持。導入はゼロ侵入。
- 使い方はシンプル。
pip install sll-core、訓練ループをwith sll.linearize():で囲むだけ。 - 対象は sign/round/floor/argmax など、微分不可能な操作を含むモデル
- 従来の STE/Gumbel-Softmax 互換性を保ちながら、モデルの大幅な改変を不要にします。
投票数: 0