def testApplyCmvn(self): #pylint: disable=invalid-name ''' test apply cmvn ''' np.random.seed(12) tf.set_random_seed(12) feat_size = 40 delta_deltas = True feat_shape = [2, 10, feat_size, 3 if delta_deltas else 1] feat = np.random.randn(*feat_shape) feat = feat.astype(np.float32) feat = tf.constant(feat) mean = feat / 2 var = feat / 3 eps = 1e-9 feat_out = utils.apply_cmvn(feat, mean, var, epsilon=eps) feat_true = (feat - mean) * tf.rsqrt(var + eps) with self.cached_session(use_gpu=False, force_gpu=False): self.assertAllClose(feat_out.eval(), feat_true.eval())
def testApplyLocalCmvn(self): #pylint: disable=invalid-name ''' test apply_local_cmvn() ''' np.random.seed(12) tf.set_random_seed(12) feat_size = 40 delta_deltas = True feat_shape = [2, 10, feat_size, 3 if delta_deltas else 1] feat = np.random.randn(*feat_shape) feat = feat.astype(np.float32) mean = np.mean(feat, axis=1, keepdims=True) var = np.var(feat, axis=1, keepdims=True) eps = 1e-9 feat_true = (feat - mean) / np.sqrt(var + eps) feat = tf.constant(feat) feat_out = utils.apply_local_cmvn(feat, epsilon=eps) with self.cached_session(use_gpu=False, force_gpu=False): self.assertAllClose(feat_out.eval(), feat_true)
def set_seed(config): """Set the random seed.""" random_seed = config['solver']['run_config']['tf_random_seed'] random.seed(random_seed) np.random.seed(random_seed) tf.set_random_seed(random_seed)