From 8beaf3279b705bfd288dac92b603a1c8bf366f31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Wed, 4 Oct 2023 21:47:43 +0800 Subject: [PATCH] Update models.py --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 18dff963..dd9e0c08 100644 --- a/models.py +++ b/models.py @@ -763,7 +763,7 @@ def __init__( gin_channels=256, use_sdp=True, n_flow_layer=4, - n_layers_trans_flow=3, + n_layers_trans_flow=6, flow_share_parameter=False, use_transformer_flow=True, **kwargs