def rand_init(self):
"""
random initialization
"""
if 'maMulVecCrowd' in self.task:
self.maMulVecCrowd.data.fill_(1)
if 'maAddVecCrowd' in self.task:
self.maAddVecCrowd.data.fill_(0)
if 'maCatVecCrowd' in self.task:
self.maCatVecCrowd.data.fill_(0)
self.maCatVecCrowd_latent.data.fill_(0)
if 'maMulMatCrowd' in self.task:
for i in range(self.a_num):
nn.init.eye_(self.maMulMatCrowd[i])
if 'maMulCRFCrowd' in self.task:
for i in range(self.a_num):
nn.init.eye_(self.maMulCRFCrowd[i])
if 'maMulScoreCrowd' in self.task:
for i in range(self.a_num):
nn.init.eye_(self.maMulScoreCrowd[i])
utils.init_linear(self.hidden2tag)
self.attention.data.zero_()
if not (('maMulCRFCrowd' in self.task) and
('latent' not in self.task)):
self.transitions.data.zero_()
def rand_init(self):
"""
Random initialization.
"""
utils.init_embedding(self.char_embed.weight)
utils.init_lstm(self.char_fw)
utils.init_lstm(self.char_bw)
utils.init_lstm(self.word_rnn)
utils.init_linear(self.char_seq)
def rand_init(self):
"""
random initialization
"""
utils.init_linear(self.hidden2tag)
self.transitions.data.zero_()
