Python init_with_uniform Examples

Example #1

File: rnn.py Project: 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)

Example #2

File: sequence_summary.py Project: 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)