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)
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)
