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