def init_weights(self): """Initializes params uniformly.""" modules.init_rnn(self.sel_rnn, self.args.init_range) #, bidirectional=True) self.word_encoder.weight.data.uniform_(-self.args.init_range, self.args.init_range) modules.init_cont([self.sel_encoder], self.args.init_range) modules.init_cont([self.sel_decoder], self.args.init_range)
def init_weights(self): self.decoder.weight.data.uniform_(-self.args.init_range, self.args.init_range) self.decoder.bias.data.fill_(0) modules.init_rnn(self.reader, self.args.init_range) self.word_encoder.weight.data.uniform_(-self.args.init_range, self.args.init_range) modules.init_cont(self.attn, self.args.init_range) modules.init_cont(self.sel_encoder, self.args.init_range) modules.init_cont(self.sel_decoders, self.args.init_range)