mirror of https://github.com/coqui-ai/TTS.git
Fix style
This commit is contained in:
parent
39db650bdb
commit
5680b599ca
|
@ -764,7 +764,7 @@ class Vits(BaseTTS):
|
||||||
orig_freq=self.config.audio["sample_rate"], new_freq=self.args.encoder_sample_rate
|
orig_freq=self.config.audio["sample_rate"], new_freq=self.args.encoder_sample_rate
|
||||||
) # pylint: disable=W0201
|
) # 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"""
|
"""Reinit layes if needed"""
|
||||||
if self.args.reinit_DP:
|
if self.args.reinit_DP:
|
||||||
before_dict = get_module_weights_sum(self.duration_predictor)
|
before_dict = get_module_weights_sum(self.duration_predictor)
|
||||||
|
|
Loading…
Reference in New Issue