espnet2.sds.end_to_end.mini_omni.litgpt.utils.chunked_cross_entropy

source

espnet2.sds.end_to_end.mini_omni.litgpt.utils.chunked_cross_entropy(logits: Tensor | List[Tensor], targets: Tensor, chunk_size: int = 128, ignore_index: int = -100) → Tensor