espnet2.uasr.segmenter.join_segmenter.JoinSegmenter

source

class espnet2.uasr.segmenter.join_segmenter.JoinSegmenter(cfg: Dict | None = None, subsample_rate: float = 0.25, mean_pool: str2bool = True, mean_join_pool: str2bool = False, remove_zeros: str2bool = False)

Bases: AbsSegmenter

Initializes internal Module state, shared by both nn.Module and ScriptModule.

logit_segment(logits: Tensor, padding_mask: Tensor) → Tensor

pre_segment(xs_pad: Tensor, padding_mask: Tensor) → Tensor