GitHub logo

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
← 投稿一覧に戻る