From 198bc583ab6c031dd9193fc9376c1f9f8c2ea655 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sat, 23 Sep 2023 21:10:37 +0800 Subject: [PATCH] decrease layers --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index dd9e0c0..f392136 100644 --- a/models.py +++ b/models.py @@ -763,7 +763,7 @@ class SynthesizerTrn(nn.Module): gin_channels=256, use_sdp=True, n_flow_layer=4, - n_layers_trans_flow=6, + n_layers_trans_flow=4, flow_share_parameter=False, use_transformer_flow=True, **kwargs