From 567a136c0f3bcb52436c70edbb0ec605f9b6c85e Mon Sep 17 00:00:00 2001 From: tyanz Date: Wed, 28 Aug 2024 18:58:30 +0800 Subject: [PATCH] change dropout_rate to float type --- .../cosyvoice/conf/cosyvoice.fromscratch.yaml | 6 +++--- examples/magicdata-read/cosyvoice/conf/cosyvoice.yaml | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/magicdata-read/cosyvoice/conf/cosyvoice.fromscratch.yaml b/examples/magicdata-read/cosyvoice/conf/cosyvoice.fromscratch.yaml index 54b6e7b..0f3495f 100644 --- a/examples/magicdata-read/cosyvoice/conf/cosyvoice.fromscratch.yaml +++ b/examples/magicdata-read/cosyvoice/conf/cosyvoice.fromscratch.yaml @@ -31,7 +31,7 @@ llm: !new:cosyvoice.llm.llm.TransformerLM num_blocks: 3 dropout_rate: 0.1 positional_dropout_rate: 0.1 - attention_dropout_rate: 0 + attention_dropout_rate: 0.0 normalize_before: True input_layer: 'linear' pos_enc_layer_type: 'rel_pos_espnet' @@ -49,7 +49,7 @@ llm: !new:cosyvoice.llm.llm.TransformerLM num_blocks: 7 dropout_rate: 0.1 positional_dropout_rate: 0.1 - attention_dropout_rate: 0 + attention_dropout_rate: 0.0 input_layer: 'linear_legacy' pos_enc_layer_type: 'rel_pos_espnet' selfattention_layer_type: 'rel_selfattn' @@ -97,7 +97,7 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec in_channels: 320 out_channels: 80 channels: [256, 256] - dropout: 0 + dropout: 0.0 attention_head_dim: 64 n_blocks: 4 num_mid_blocks: 8 diff --git a/examples/magicdata-read/cosyvoice/conf/cosyvoice.yaml b/examples/magicdata-read/cosyvoice/conf/cosyvoice.yaml index f43af16..6b084f2 100644 --- a/examples/magicdata-read/cosyvoice/conf/cosyvoice.yaml +++ b/examples/magicdata-read/cosyvoice/conf/cosyvoice.yaml @@ -31,7 +31,7 @@ llm: !new:cosyvoice.llm.llm.TransformerLM num_blocks: 6 dropout_rate: 0.1 positional_dropout_rate: 0.1 - attention_dropout_rate: 0 + attention_dropout_rate: 0.0 normalize_before: True input_layer: 'linear' pos_enc_layer_type: 'rel_pos_espnet' @@ -49,7 +49,7 @@ llm: !new:cosyvoice.llm.llm.TransformerLM num_blocks: 14 dropout_rate: 0.1 positional_dropout_rate: 0.1 - attention_dropout_rate: 0 + attention_dropout_rate: 0.0 input_layer: 'linear_legacy' pos_enc_layer_type: 'rel_pos_espnet' selfattention_layer_type: 'rel_selfattn' @@ -97,7 +97,7 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec in_channels: 320 out_channels: 80 channels: [256, 256] - dropout: 0 + dropout: 0.0 attention_head_dim: 64 n_blocks: 4 num_mid_blocks: 12