Fix style

This commit is contained in:
Edresson Casanova 2022-05-02 15:51:42 -03:00
parent 39db650bdb
commit 5680b599ca
1 changed files with 2 additions and 2 deletions

View File

@ -764,7 +764,7 @@ class Vits(BaseTTS):
orig_freq=self.config.audio["sample_rate"], new_freq=self.args.encoder_sample_rate
) # pylint: disable=W0201
def on_init_end(self, trainer): # pylint: disable=W0613
def on_init_end(self, trainer): # pylint: disable=W0613
"""Reinit layes if needed"""
if self.args.reinit_DP:
before_dict = get_module_weights_sum(self.duration_predictor)