mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
fixed script for inference
This commit is contained in:
parent
93eedced80
commit
b65eba2d15
@ -289,9 +289,9 @@ 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()])
|
||||
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()])
|
||||
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()]
|
||||
|
Loading…
x
Reference in New Issue
Block a user