mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-04 06:34:20 +00:00
Update train.py
This commit is contained in:
parent
2eb06451ee
commit
523635cf77
@ -279,7 +279,7 @@ def get_model(params: AttributeDict) -> nn.Module:
|
||||
"spks": params.num_spks,
|
||||
"langs": None,
|
||||
"spk_embed_dim": None,
|
||||
"global_channels": -1,
|
||||
"global_channels": 256,
|
||||
"segment_size": 32,
|
||||
"text_encoder_attention_heads": 2,
|
||||
"text_encoder_ffn_expand": 4,
|
||||
|
Loading…
x
Reference in New Issue
Block a user