def _propagate(self, Y, dropout=0.):
relu_1 = F.clipped_relu(self.relu_1(Y, dropout=dropout))
relu_2 = F.clipped_relu(self.relu_2(relu_1, dropout=dropout))
relu_3 = F.clipped_relu(self.relu_3(relu_2, dropout=dropout))
N_mask = F.sigmoid(self.noise_mask_estimate(relu_3))
X_mask = F.sigmoid(self.speech_mask_estimate(relu_3))
return N_mask, X_mask
评论列表
文章目录