From 43267e3e29dcc68c89ec8c5d1fdd192832216fda Mon Sep 17 00:00:00 2001 From: JinZr Date: Tue, 8 Oct 2024 13:12:12 +0800 Subject: [PATCH] black formatted --- egs/libritts/CODEC/encodec/infer.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/egs/libritts/CODEC/encodec/infer.py b/egs/libritts/CODEC/encodec/infer.py index 6be869534..2abe2493b 100755 --- a/egs/libritts/CODEC/encodec/infer.py +++ b/egs/libritts/CODEC/encodec/infer.py @@ -289,9 +289,17 @@ def main(): logging.info(f"Number of parameters in decoder: {num_param_d}") num_param_q = sum([p.numel() for p in quantizer.parameters()]) logging.info(f"Number of parameters in quantizer: {num_param_q}") - num_param_ds = sum([p.numel() for p in multi_scale_discriminator.parameters()]) if multi_scale_discriminator is not None else 0 + num_param_ds = ( + sum([p.numel() for p in multi_scale_discriminator.parameters()]) + if multi_scale_discriminator is not None + else 0 + ) logging.info(f"Number of parameters in multi_scale_discriminator: {num_param_ds}") - num_param_dp = sum([p.numel() for p in multi_period_discriminator.parameters()]) if multi_period_discriminator is not None else 0 + num_param_dp = ( + sum([p.numel() for p in multi_period_discriminator.parameters()]) + if multi_period_discriminator is not None + else 0 + ) logging.info(f"Number of parameters in multi_period_discriminator: {num_param_dp}") num_param_dstft = sum( [p.numel() for p in multi_scale_stft_discriminator.parameters()]