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)