Copy as Markdown View as Markdown

tinker.types.CustomLossInput

Generated from tinker 0.38.0 at commit 4fb85c7. Source links point at that snapshot.

class tinker.types.CustomLossInput(StrictBase)

One input of a custom loss, with the rank every datum's value has.

Fields:

• name (str) – logprobs, or the loss_fn_inputs key the loss reads.
• rank (CustomLossInputRank)

Referenced by

• tinker.types.CreateCustomLossRequest