Tinkex.Regularizers.GradientPenalty (Tinkex v0.3.4)

View Source

Gradient penalty adapter built on NxPenalties primitives.

Modes:

  • :output - Penalize gradient norm w.r.t. current logprobs
  • :interpolated - WGAN-GP style penalty between logprobs and a reference