MCPcopy Create free account
hub / github.com/Xflick/EEND_PyTorch / __init__

Method __init__

eend/pytorch_backend/models.py:47–75  ·  view source on GitHub ↗

Self-attention-based diarization model. Args: n_speakers (int): Number of speakers in recording in_size (int): Dimension of input feature vector n_heads (int): Number of attention heads n_units (int): Number of units in a self-attention block

(self, n_speakers, in_size, n_heads, n_units, n_layers, dim_feedforward=2048, dropout=0.5, has_pos=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

init_weightsMethod · 0.95
PositionalEncodingClass · 0.85
__init__Method · 0.45

Tested by

no test coverage detected