models/parallel_wavegan.py [697:720]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            stats += np.array(
                get_model_complexity_info(
                    model.upsample_net,
                    ([spectrograms]),
                    custom_modules_hooks=custom_modules_hooks,
                )
            )
            spectrograms = model.upsample_net(spectrograms)
            assert spectrograms.size(-1) == waveforms.size(-1)

            # encode to hidden representation
            stats += np.array(
                get_model_complexity_info(
                    model.first_conv,
                    ([waveforms]),
                    custom_modules_hooks=custom_modules_hooks,
                )
            )
            waveforms = model.first_conv(waveforms)
            skips = 0
            for f in model.conv_layers:
                stats += np.array(
                    get_model_complexity_info(
                        f,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models/wavenet.py [467:492]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            stats += np.array(
                get_model_complexity_info(
                    model.upsample_net,
                    ([spectrograms]),
                    custom_modules_hooks=custom_modules_hooks,
                )
            )

            spectrograms = model.upsample_net(spectrograms)
            assert spectrograms.size(-1) == waveforms.size(-1)

            stats += np.array(
                get_model_complexity_info(
                    model.first_conv,
                    ([waveforms]),
                    custom_modules_hooks=custom_modules_hooks,
                )
            )

            waveforms = model.first_conv(waveforms)
            skips = 0

            for f in model.conv_layers:
                stats += np.array(
                    get_model_complexity_info(
                        f,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -