Python init_with_uniform Exemples

Exemple #1

Fichier : rnn.py Projet : xin-w8023/neural_sp

 def reset_parameters(self, param_init):
     """Initialize parameters with uniform distribution."""
     logger.info('===== Initialize %s with uniform distribution =====' % self.__class__.__name__)
     for n, p in self.named_parameters():
         if 'conv' in n or 'tds' in n or 'gated_conv' in n:
             continue  # for CNN layers before RNN layers
         init_with_uniform(n, p, param_init)

Exemple #2

Fichier : sequence_summary.py Projet : zh794390558/neural_sp

 def reset_parameters(self, param_init):
     """Initialize parameters with uniform distribution."""
     logger.info('===== Initialize %s with uniform distribution =====' %
                 self.__class__.__name__)
     for n, p in self.named_parameters():
         init_with_uniform(n, p, param_init)