Python init_with_uniform Beispiele

Beispiel #1

Datei: rnn.py Projekt: 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)

Beispiel #2

Datei: sequence_summary.py Projekt: 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)