From a196c052cb7f7ca4efed4193229bcabd2e56d29c Mon Sep 17 00:00:00 2001 From: co63oc Date: Fri, 17 Jan 2025 17:00:47 +0800 Subject: [PATCH] Fix (#3974) --- tests/unit/tts/test_pwg.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/tts/test_pwg.py b/tests/unit/tts/test_pwg.py index 10c82c9fd..bcdb5aafc 100644 --- a/tests/unit/tts/test_pwg.py +++ b/tests/unit/tts/test_pwg.py @@ -14,16 +14,16 @@ import paddle import torch from paddle.device.cuda import synchronize +from parallel_wavegan import models as pwgan from parallel_wavegan.layers import residual_block from parallel_wavegan.layers import upsample -from parallel_wavegan.models import parallel_wavegan as pwgan from timer import timer from paddlespeech.t2s.models.parallel_wavegan import ConvInUpsampleNet from paddlespeech.t2s.models.parallel_wavegan import PWGDiscriminator from paddlespeech.t2s.models.parallel_wavegan import PWGGenerator -from paddlespeech.t2s.models.parallel_wavegan import ResidualBlock from paddlespeech.t2s.models.parallel_wavegan import ResidualPWGDiscriminator +from paddlespeech.t2s.modules.residual_block import WaveNetResidualBlock from paddlespeech.t2s.utils.layer_tools import summary paddle.set_device("gpu:0") @@ -79,8 +79,8 @@ def test_convin_upsample_net(): def test_residual_block(): - net = ResidualBlock(dilation=9) - net2 = residual_block.ResidualBlock(dilation=9) + net = WaveNetResidualBlock(dilation=9) + net2 = residual_block.WaveNetResidualBlock(dilation=9) summary(net) summary(net2) for k, v in net2.named_parameters():