← Kembali ke katalog
G

GitHub

Differentiable sign/round/floor for PyTorch

4.5(10,000 ulasan)
Gratis
Diluncurkan 2026

Sobre

PyTorch's sign, round, floor, and argmax break gradients. Existing fixes (STE, Gumbel-Softmax) require rewriting your model. SLL-Core is zero-intrusive: with sll.linearize(): y = torch.sign(x) # now differentiable! loss.backward() Only linearizes inside an ε-band around decision boundaries. Everywhere else stays exactly hard. Zero deployment overhead. pip install sll-core

Kelebihan

  • +Zero deployment overhead
  • +Não requer reescrita do modelo
  • +Suporte a gradientes

Kekurangan

  • Limitado a ε-band em torno de decisões de fronteira
  • Não claro se é adequado para todos os casos

Você também pode gostar