zhifu gao
2022-12-09 f9fed09e96f43e7eab88378fc444c4987933badb
egs/aishell/tranformer/utils/compute_cmvn.py
@@ -11,6 +11,13 @@
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )
    parser.add_argument(
        "--dims",
        "-d",
        default=80,
        type=int,
        help="feature dims",
    )
    parser.add_argument(
        "--ark-file",
        "-a",
        default=False,
@@ -44,8 +51,8 @@
    ark_file = args.ark_file + "/feats." + str(args.ark_index) + ".ark"
    cmvn_file = args.output_dir + "/cmvn." + str(args.ark_index) + ".json"
    mean_stats = 0.0
    var_stats = 0.0
    mean_stats = np.zeros(args.dims)
    var_stats = np.zeros(args.dims)
    total_frames = 0
    with ReadHelper('ark:{}'.format(ark_file)) as ark_reader: