Function trim_batch

seq2seq/utils.py:77–87 · view source on GitHub ↗

Remove columns that are populated exclusively by pad_token_id

(
    input_ids,
    pad_token_id,
    attention_mask=None,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

collate_fnMethod · 0.85

no outgoing calls

no test coverage detected