def __init__(self, in_size, out_size, trainer): self.out_size_ = out_size self.pwxh_ = Parameter([4 * out_size, in_size], I.Uniform(-0.1, 0.1)) self.pwhh_ = Parameter([4 * out_size, out_size], I.Uniform(-0.1, 0.1)) self.pbh_ = Parameter([4 * out_size], I.Constant(0)) trainer.add_parameter(self.pwxh_) trainer.add_parameter(self.pwhh_) trainer.add_parameter(self.pbh_)
def __init__(self, vocab_size, eos_id): self.eos_id = eos_id self.plookup = Parameter([NUM_HIDDEN_UNITS, vocab_size], I.Uniform(-0.1, 0.1)) self.rnn1 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS) self.rnn2 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS) self.hy = Affine(NUM_HIDDEN_UNITS, vocab_size) self.scan_attributes()
def __init__(self, in_size, out_size, trainer): self.out_size_ = out_size self.pw_ = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1)) self.pbf_ = Parameter([out_size], I.Constant(0)) self.pbr_ = Parameter([out_size], I.Constant(0)) trainer.add_parameter(self.pw_) trainer.add_parameter(self.pbf_) trainer.add_parameter(self.pbr_)
def __init__(self, vocab_size, eos_id, trainer): self.eos_id_ = eos_id self.plookup_ = Parameter([NUM_HIDDEN_UNITS, vocab_size], I.Uniform(-0.1, 0.1)) self.rnn1_ = SRU(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS, trainer) self.rnn2_ = SRU(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS, trainer) self.hy_ = Affine(NUM_HIDDEN_UNITS, vocab_size, trainer) trainer.add_parameter(self.plookup_)
def __init__(self, vocab_size, eos_id): self.eos_id = eos_id self.plookup = Parameter([NUM_HIDDEN_UNITS, vocab_size], I.Uniform(-0.1, 0.1)) self.rnn1 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS) self.rnn2 = LSTM(NUM_HIDDEN_UNITS, NUM_HIDDEN_UNITS) self.hy = Affine(NUM_HIDDEN_UNITS, vocab_size) self.add_all_parameters() self.add_all_submodels()
def __init__(self, in_size, out_size): self.out_size = out_size self.pw = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1)) self.pbf = Parameter([out_size], I.Constant(0)) self.pbr = Parameter([out_size], I.Constant(0)) self.scan_attributes()
def __init__(self, in_size, out_size): self.out_size = out_size self.pw = Parameter([3 * out_size, in_size], I.Uniform(-0.1, 0.1)) self.pbf = Parameter([out_size], I.Constant(0)) self.pbr = Parameter([out_size], I.Constant(0)) self.add_all_parameters()