MCPcopy
hub / github.com/FareedKhan-dev/train-llm-from-scratch / _logps

Function _logps

scripts/train_dpo.py:33–34  ·  view source on GitHub ↗
(model, ids, mask, requires_grad)

Source from the content-addressed store, hash-verified

31
32
33def _logps(model, ids, mask, requires_grad):
34 return sequence_logprobs(model, ids, mask, requires_grad=requires_grad)
35
36
37def _compute_losses(policy, ref, batch, cfg, ctx):

Callers 1

_compute_lossesFunction · 0.85

Calls 1

sequence_logprobsFunction · 0.90

Tested by

no test coverage detected