SYMBOL INDEX (3493 symbols across 424 files) FILE: asr/asr_mix_utils.py class PlotAttentionReport (line 30) | class PlotAttentionReport(extension.Extension): method __init__ (line 45) | def __init__(self, att_vis_fn, data, outdir, converter, device, revers... method __call__ (line 56) | def __call__(self, trainer): method log_attentions (line 69) | def log_attentions(self, logger, step): method get_attention_weights (line 79) | def get_attention_weights(self): method get_attention_weight (line 94) | def get_attention_weight(self, idx, att_w, spkr_idx): method draw_attention_plot (line 108) | def draw_attention_plot(self, att_w): method _plot_and_save_attention (line 133) | def _plot_and_save_attention(self, att_w, filename): function add_results_to_json (line 139) | def add_results_to_json(js, nbest_hyps_sd, char_list): FILE: asr/asr_utils.py class CompareValueTrigger (line 18) | class CompareValueTrigger(object): method __init__ (line 28) | def __init__(self, key, compare_fn, trigger=(1, "epoch")): method __call__ (line 37) | def __call__(self, trainer): method _init_summary (line 62) | def _init_summary(self): class PlotAttentionReport (line 74) | class PlotAttentionReport(extension.Extension): method __init__ (line 98) | def __init__( method __call__ (line 130) | def __call__(self, trainer): method log_attentions (line 180) | def log_attentions(self, logger, step): method get_attention_weights (line 210) | def get_attention_weights(self): method trim_attention_weight (line 229) | def trim_attention_weight(self, uttid, att_w): method draw_attention_plot (line 247) | def draw_attention_plot(self, att_w): method draw_han_plot (line 274) | def draw_han_plot(self, att_w): method _plot_and_save_attention (line 314) | def _plot_and_save_attention(self, att_w, filename, han_mode=False): class PlotCTCReport (line 329) | class PlotCTCReport(extension.Extension): method __init__ (line 353) | def __init__( method __call__ (line 385) | def __call__(self, trainer): method log_ctc_probs (line 419) | def log_ctc_probs(self, logger, step): method get_ctc_probs (line 439) | def get_ctc_probs(self): method trim_ctc_prob (line 455) | def trim_ctc_prob(self, uttid, prob): method draw_ctc_plot (line 463) | def draw_ctc_plot(self, ctc_prob): method _plot_and_save_ctc (line 499) | def _plot_and_save_ctc(self, ctc_prob, filename): function restore_snapshot (line 505) | def restore_snapshot(model, snapshot, load_fn=None): function _restore_snapshot (line 525) | def _restore_snapshot(model, snapshot, load_fn=None): function adadelta_eps_decay (line 535) | def adadelta_eps_decay(eps_decay): function _adadelta_eps_decay (line 554) | def _adadelta_eps_decay(trainer, eps_decay): function adam_lr_decay (line 568) | def adam_lr_decay(eps_decay): function _adam_lr_decay (line 587) | def _adam_lr_decay(trainer, eps_decay): function torch_snapshot (line 601) | def torch_snapshot(savefun=torch.save, filename="snapshot.ep.{.updater.e... function _torch_snapshot_object (line 617) | def _torch_snapshot_object(trainer, target, filename, savefun): function add_gradient_noise (line 661) | def add_gradient_noise(model, iteration, duration=100, eta=1.0, scale_fa... function get_model_conf (line 685) | def get_model_conf(model_path, conf_path=None): function chainer_load (line 713) | def chainer_load(path, model): function torch_save (line 729) | def torch_save(path, model): function snapshot_object (line 743) | def snapshot_object(target, filename): function torch_load (line 767) | def torch_load(path, model): function torch_resume (line 790) | def torch_resume(snapshot_path, trainer, load_trainer_and_opt=True): function parse_hypothesis (line 836) | def parse_hypothesis(hyp, char_list): function add_results_to_json (line 860) | def add_results_to_json(js, nbest_hyps, char_list): function plot_spectrogram (line 921) | def plot_spectrogram( function format_mulenc_args (line 1001) | def format_mulenc_args(args): FILE: asr/chainer_backend/asr.py function train (line 51) | def train(args): function recog (line 480) | def recog(args): FILE: asr/pytorch_backend/asr.py function _recursive_to (line 77) | def _recursive_to(xs, device): function is_alphabet (line 84) | def is_alphabet(char): class CustomEvaluator (line 90) | class CustomEvaluator(BaseEvaluator): method __init__ (line 106) | def __init__(self, model, iterator, target, device, ngpu=None): method evaluate (line 118) | def evaluate(self): class CustomUpdater (line 155) | class CustomUpdater(StandardUpdater): method __init__ (line 170) | def __init__( method update_core (line 197) | def update_core(self): method update (line 261) | def update(self): class CustomConverter (line 269) | class CustomConverter(object): method __init__ (line 278) | def __init__(self, subsampling_factor=1, dtype=torch.float32): method __call__ (line 284) | def __call__(self, batch, device=torch.device("cpu")): class CustomConverterMulEnc (line 344) | class CustomConverterMulEnc(object): method __init__ (line 353) | def __init__(self, subsamping_factors=[1, 1], dtype=torch.float32): method __call__ (line 360) | def __call__(self, batch, device=torch.device("cpu")): function train (line 412) | def train(args): function recog (line 1016) | def recog(args): function enhance (line 1368) | def enhance(args): function ctc_align (line 1587) | def ctc_align(args): FILE: asr/pytorch_backend/asr_init.py function freeze_modules (line 18) | def freeze_modules(model, modules): function transfer_verification (line 40) | def transfer_verification(model_state_dict, partial_state_dict, modules): function get_partial_state_dict (line 72) | def get_partial_state_dict(model_state_dict, modules): function get_lm_state_dict (line 94) | def get_lm_state_dict(lm_state_dict): function filter_modules (line 118) | def filter_modules(model_state_dict, modules): function load_trained_model (line 151) | def load_trained_model(model_path, training=True): function _load_trained_model (line 185) | def _load_trained_model(model_path, training=True, patience=10): function get_trained_model_state_dict (line 195) | def get_trained_model_state_dict(model_path): function load_trained_modules (line 233) | def load_trained_modules(idim, odim, args, interface=ASRInterface): FILE: asr/pytorch_backend/asr_mix.py class CustomConverter (line 54) | class CustomConverter(object): method __init__ (line 63) | def __init__(self, subsampling_factor=1, dtype=torch.float32, num_spkr... method __call__ (line 70) | def __call__(self, batch, device=torch.device("cpu")): function train (line 130) | def train(args): function recog (line 532) | def recog(args): FILE: asr/pytorch_backend/recog.py function recog_v2 (line 27) | def recog_v2(args): FILE: bin/asr_align.py function get_parser (line 64) | def get_parser(): function main (line 184) | def main(args): function ctc_align (line 228) | def ctc_align(args, device): FILE: bin/asr_enhance.py function get_parser (line 15) | def get_parser(): function main (line 138) | def main(args): FILE: bin/asr_recog.py function get_parser (line 22) | def get_parser(): function main (line 373) | def main(args): FILE: bin/asr_train.py function get_parser (line 29) | def get_parser(parser=None, required=True): function main (line 576) | def main(cmd_args): FILE: bin/lm_train.py function get_parser (line 27) | def get_parser(parser=None, required=True): function main (line 190) | def main(cmd_args): FILE: bin/mt_train.py function get_parser (line 29) | def get_parser(parser=None, required=True): function main (line 375) | def main(cmd_args): FILE: bin/mt_trans.py function get_parser (line 19) | def get_parser(): function main (line 125) | def main(args): FILE: bin/st_train.py function get_parser (line 28) | def get_parser(parser=None, required=True): function main (line 445) | def main(cmd_args): FILE: bin/st_trans.py function get_parser (line 19) | def get_parser(): function main (line 122) | def main(args): FILE: bin/tts_decode.py function get_parser (line 19) | def get_parser(): function main (line 118) | def main(args): FILE: bin/tts_train.py function get_parser (line 24) | def get_parser(): function main (line 294) | def main(cmd_args): FILE: bin/vc_decode.py function get_parser (line 19) | def get_parser(): function main (line 112) | def main(args): FILE: bin/vc_train.py function get_parser (line 24) | def get_parser(): function main (line 304) | def main(cmd_args): FILE: egs/aishell1/local/make_lexicon_fst.py function get_args (line 21) | def get_args(): function read_lexiconp (line 60) | def read_lexiconp(filename): function write_nonterminal_arcs (line 119) | def write_nonterminal_arcs(start_state, loop_state, next_state, function write_fst_no_silence (line 173) | def write_fst_no_silence(lexicon, nonterminals=None, left_context_phones... function write_fst_with_silence (line 220) | def write_fst_with_silence(lexicon, sil_prob, sil_phone, sil_disambig, function write_words_txt (line 308) | def write_words_txt(orig_lines, highest_numbered_symbol, nonterminals, f... function read_nonterminals (line 322) | def read_nonterminals(filename): function read_left_context_phones (line 338) | def read_left_context_phones(filename): function is_token (line 354) | def is_token(s): function main (line 363) | def main(): FILE: egs/aishell2/local/make_lexicon_fst.py function get_args (line 21) | def get_args(): function read_lexiconp (line 60) | def read_lexiconp(filename): function write_nonterminal_arcs (line 119) | def write_nonterminal_arcs(start_state, loop_state, next_state, function write_fst_no_silence (line 173) | def write_fst_no_silence(lexicon, nonterminals=None, left_context_phones... function write_fst_with_silence (line 220) | def write_fst_with_silence(lexicon, sil_prob, sil_phone, sil_disambig, function write_words_txt (line 308) | def write_words_txt(orig_lines, highest_numbered_symbol, nonterminals, f... function read_nonterminals (line 322) | def read_nonterminals(filename): function read_left_context_phones (line 338) | def read_left_context_phones(filename): function is_token (line 354) | def is_token(s): function main (line 363) | def main(): FILE: egs/asrucs/local/add_seperator.py function is_all_chinese (line 3) | def is_all_chinese(strs): FILE: egs/asrucs/local/generate_fake_cs.py function read_datadir (line 9) | def read_datadir(d): function generate_pairs (line 27) | def generate_pairs(chn_dur_dict, eng_dur_dict, dur_maximum): function write_utts (line 51) | def write_utts(tgt_dir, pair_list, wav_dict, text_dict): function main (line 81) | def main(): FILE: egs/espnet_utils/add_uttcls_json.py function main (line 4) | def main(): FILE: egs/espnet_utils/addjson.py function get_parser (line 21) | def get_parser(): FILE: egs/espnet_utils/apply-cmvn.py function get_parser (line 16) | def get_parser(): function main (line 102) | def main(): FILE: egs/espnet_utils/average_checkpoints.py function main (line 11) | def main(): function get_parser (line 120) | def get_parser(): FILE: egs/espnet_utils/calculate_rtf.py function get_parser (line 14) | def get_parser(): function main (line 25) | def main(): FILE: egs/espnet_utils/change_yaml.py function get_parser (line 8) | def get_parser(): function main (line 35) | def main(): FILE: egs/espnet_utils/compute-cmvn-stats.py function get_parser (line 15) | def get_parser(): function main (line 63) | def main(): FILE: egs/espnet_utils/compute-fbank-feats.py function get_parser (line 20) | def get_parser(): function main (line 92) | def main(): FILE: egs/espnet_utils/compute-stft-feats.py function get_parser (line 20) | def get_parser(): function main (line 84) | def main(): FILE: egs/espnet_utils/concat_json_multiref.py function get_parser (line 16) | def get_parser(): function main (line 25) | def main(): FILE: egs/espnet_utils/concatjson.py function get_parser (line 19) | def get_parser(): function truncate_tail (line 29) | def truncate_tail(d, size): FILE: egs/espnet_utils/convert_fbank_to_wav.py function logmelspc_to_linearspc (line 23) | def logmelspc_to_linearspc(lmspc, fs, n_mels, n_fft, fmin=None, fmax=None): function griffin_lim (line 49) | def griffin_lim(spc, n_fft, n_shift, win_length, window="hann", n_iters=... function get_parser (line 98) | def get_parser(): function main (line 147) | def main(): FILE: egs/espnet_utils/copy-feats.py function get_parser (line 13) | def get_parser(): function main (line 63) | def main(): FILE: egs/espnet_utils/dump-pcm.py function get_parser (line 14) | def get_parser(): function main (line 81) | def main(): FILE: egs/espnet_utils/eval-source-separation.py function eval_STOI (line 23) | def eval_STOI(ref, y, fs, extended=False, compute_permutation=True): function eval_PESQ (line 75) | def eval_PESQ(ref, enh, fs, compute_permutation: bool = True, wideband: ... function get_parser (line 178) | def get_parser(): function main (line 244) | def main(): FILE: egs/espnet_utils/eval_perm_free_error.py function permutationDFS (line 17) | def permutationDFS(source, start, res): function permutation_schemes (line 32) | def permutation_schemes(num_spkrs): function convert_score (line 50) | def convert_score(keys, dic): function get_utt_permutation (line 61) | def get_utt_permutation(old_dic, num_spkrs=2): function get_results (line 90) | def get_results(result_file, result_key): function merge_results (line 129) | def merge_results(results): function get_parser (line 161) | def get_parser(): function main (line 177) | def main(): FILE: egs/espnet_utils/feat-to-shape.py function get_parser (line 12) | def get_parser(): function main (line 45) | def main(): FILE: egs/espnet_utils/feats2npy.py function get_parser (line 12) | def get_parser(): FILE: egs/espnet_utils/filt.py function get_parser (line 12) | def get_parser(): function main (line 29) | def main(args): function filter_file (line 34) | def filter_file(infile, filt, exclude): FILE: egs/espnet_utils/filter_all_eng_utts.py function is_all_chinese (line 3) | def is_all_chinese(strs): FILE: egs/espnet_utils/generate_wav_from_fbank.py class TimeInvariantMLSAFilter (line 29) | class TimeInvariantMLSAFilter(object): method __init__ (line 47) | def __init__(self, coef, alpha, n_shift): method __call__ (line 55) | def __call__(self, y): function get_parser (line 76) | def get_parser(): function main (line 100) | def main(): FILE: egs/espnet_utils/get_yaml.py function get_parser (line 7) | def get_parser(): function main (line 19) | def main(): FILE: egs/espnet_utils/json2sctm.py function get_parser (line 13) | def get_parser(): function main (line 31) | def main(args): function wrd_name (line 96) | def wrd_name(trn): FILE: egs/espnet_utils/json2text.py function get_parser (line 15) | def get_parser(): FILE: egs/espnet_utils/json2trn.py function get_parser (line 17) | def get_parser(): function main (line 30) | def main(args): function convert (line 35) | def convert(jsonf, dic, refs, hyps, num_spkrs=1): FILE: egs/espnet_utils/json2trn_mt.py function get_parser (line 18) | def get_parser(): function main (line 38) | def main(args): function convert (line 44) | def convert(jsonf, dic, refs, hyps, srcs, dic_src): FILE: egs/espnet_utils/json2trn_wo_dict.py function get_parser (line 16) | def get_parser(): function main (line 28) | def main(args): function convert (line 33) | def convert(jsonf, refs, hyps, num_spkrs=1): FILE: egs/espnet_utils/make_pair_json.py function get_parser (line 16) | def get_parser(): FILE: egs/espnet_utils/mbr_analysis.py function main (line 11) | def main(): FILE: egs/espnet_utils/mcd_calculate.py function find_files (line 24) | def find_files(root_dir, query="*.wav", include_root_dir=True): function low_cut_filter (line 46) | def low_cut_filter(x, fs, cutoff=70): function spc2npow (line 68) | def spc2npow(spectrogram): function _spvec2pow (line 92) | def _spvec2pow(specvec): function extfrm (line 120) | def extfrm(data, npow, power_threshold=-20): function world_extract (line 152) | def world_extract(wav_path, args): function get_basename (line 175) | def get_basename(path): function calculate (line 179) | def calculate(file_list, gt_file_list, args, MCD): function get_parser (line 213) | def get_parser(): function main (line 251) | def main(): FILE: egs/espnet_utils/merge_scp2json.py function shape (line 21) | def shape(x): function get_parser (line 41) | def get_parser(): FILE: egs/espnet_utils/mergejson.py function get_parser (line 20) | def get_parser(): FILE: egs/espnet_utils/mix-mono-wav-scp.py function get_parser (line 14) | def get_parser(): function main (line 30) | def main(): FILE: egs/espnet_utils/prepare_mer.py function is_all_chinese (line 3) | def is_all_chinese(strs): FILE: egs/espnet_utils/result2json.py function get_parser (line 17) | def get_parser(): FILE: egs/espnet_utils/score_lang_id.py function get_parser (line 12) | def get_parser(): function main (line 25) | def main(args): function scoring (line 30) | def scoring(ref, hyp, out): FILE: egs/espnet_utils/scp2json.py function get_parser (line 15) | def get_parser(): FILE: egs/espnet_utils/splitjson.py function get_parser (line 20) | def get_parser(): FILE: egs/espnet_utils/text2token.py function exist_or_not (line 15) | def exist_or_not(i, match_pos): function get_parser (line 27) | def get_parser(): function main (line 68) | def main(): FILE: egs/espnet_utils/text2vocabulary.py function get_parser (line 15) | def get_parser(): FILE: egs/espnet_utils/text_norm.py function remove_punc (line 13) | def remove_punc(s): function is_all_chinese (line 18) | def is_all_chinese(strs): function is_contain_chinese (line 24) | def is_contain_chinese(check_str): function splice (line 30) | def splice(s): function digit_norm (line 63) | def digit_norm(s): function remove_blank_chn (line 85) | def remove_blank_chn(s): function add_blank_boundary (line 99) | def add_blank_boundary(s): function split_eng_words (line 114) | def split_eng_words(s): function split_chn_words (line 125) | def split_chn_words(s): function upper_or_lower (line 135) | def upper_or_lower(s, upper=True): function process_one_line (line 141) | def process_one_line(content, args): function get_parser (line 181) | def get_parser(): function main (line 204) | def main(): FILE: egs/espnet_utils/trace_rnnt.py function main (line 23) | def main(): FILE: egs/espnet_utils/trim_silence.py function _time_to_str (line 21) | def _time_to_str(time_idx): function get_parser (line 26) | def get_parser(): function main (line 63) | def main(): FILE: egs/espnet_utils/trn2ctm.py function get_parser (line 12) | def get_parser(): function main (line 19) | def main(args): function convert (line 24) | def convert(trn=None, ctm=None): FILE: egs/espnet_utils/trn2stm.py function get_parser (line 11) | def get_parser(): function main (line 25) | def main(args): function convert (line 30) | def convert(trn=None, stm=None, orig_stm=None): FILE: egs/espnet_utils/word_ngram_rescore.py function score_texts (line 11) | def score_texts(ngram, texts, ignore_strs=["", " "]): function main (line 16) | def main(): FILE: egs/steps/cleanup/combine_short_segments.py function GetArgs (line 16) | def GetArgs(): function RunKaldiCommand (line 34) | def RunKaldiCommand(command, wait = True): function MakeDir (line 49) | def MakeDir(dir): function CheckFiles (line 58) | def CheckFiles(input_data_dir): function ParseFileToDict (line 64) | def ParseFileToDict(file, assert2fields = False, value_processor = None): function WriteDictToFile (line 77) | def WriteDictToFile(dict, file_name): function ParseDataDirInfo (line 92) | def ParseDataDirInfo(data_dir): function GetCombinedUttIndexRange (line 108) | def GetCombinedUttIndexRange(utt_index, utts, utt_durs, minimum_duration): function WriteCombinedDirFiles (line 175) | def WriteCombinedDirFiles(output_dir, utt2spk, spk2utt, text, feat, utt2... function CombineSegments (line 242) | def CombineSegments(input_dir, output_dir, minimum_duration): function Main (line 298) | def Main(): FILE: egs/steps/cleanup/internal/align_ctm_ref.py function get_args (line 33) | def get_args(): function read_text (line 115) | def read_text(text_file): function read_ctm (line 139) | def read_ctm(ctm_file, file_and_channel2reco=None): function smith_waterman_alignment (line 183) | def smith_waterman_alignment(ref, hyp, similarity_score_function, function print_alignment (line 355) | def print_alignment(recording, alignment, out_file_handle): function get_edit_type (line 367) | def get_edit_type(hyp_word, ref_word, duration=-1, eps_symbol='', function get_ctm_edits (line 388) | def get_ctm_edits(alignment_output, ctm_array, eps_symbol="", function ctm_line_to_string (line 491) | def ctm_line_to_string(ctm_line): function test_alignment (line 499) | def test_alignment(align_full_hyp): function run (line 514) | def run(args): function main (line 609) | def main(): FILE: egs/steps/cleanup/internal/compute_tf_idf.py function _get_args (line 21) | def _get_args(): function _run (line 87) | def _run(args): function main (line 130) | def main(): FILE: egs/steps/cleanup/internal/get_ctm_edits.py function OpenFiles (line 116) | def OpenFiles(): function PadArrays (line 173) | def PadArrays(edits_array, ctm_array): function GetEditType (line 223) | def GetEditType(hyp_word, ref_word, duration): function FloatToString (line 245) | def FloatToString(f): function OutputCtm (line 255) | def OutputCtm(utterance_id, edits_array, ctm_array): function ProcessOneUtterance (line 275) | def ProcessOneUtterance(utterance_id, edits_line, ctm_lines): function ProcessData (line 319) | def ProcessData(): FILE: egs/steps/cleanup/internal/get_non_scored_words.py function read_lang (line 56) | def read_lang(lang_dir): FILE: egs/steps/cleanup/internal/get_pron_stats.py function GetArgs (line 27) | def GetArgs(): function CheckArgs (line 54) | def CheckArgs(args): function ReadEntries (line 68) | def ReadEntries(file_handle): function GetStatsFromCtmProns (line 122) | def GetStatsFromCtmProns(silphones, optional_silence, non_scored_words, ... function WriteStats (line 214) | def WriteStats(stats, file_handle): function Main (line 219) | def Main(): FILE: egs/steps/cleanup/internal/make_one_biased_lm.py class NgramCounts (line 56) | class NgramCounts(object): method __init__ (line 67) | def __init__(self, ngram_order): method AddCount (line 96) | def AddCount(self, history, predicted_word, count): method AddRawCountsFromLine (line 101) | def AddRawCountsFromLine(self, line): method AddRawCountsFromStandardInput (line 114) | def AddRawCountsFromStandardInput(self): method GetHistToTotalCount (line 131) | def GetHistToTotalCount(self): method CompletelyDiscountLowCountStates (line 146) | def CompletelyDiscountLowCountStates(self, min_count): method ApplyBackoff (line 165) | def ApplyBackoff(self, D): method Print (line 185) | def Print(self, info_string): method AddTopWords (line 205) | def AddTopWords(self, top_words_file): method GetTotalCountMap (line 230) | def GetTotalCountMap(self): method GetHistToStateMap (line 239) | def GetHistToStateMap(self): method GetProb (line 250) | def GetProb(self, hist, word, total_count_map): method PrintAsFst (line 261) | def PrintAsFst(self, word_disambig_symbol): FILE: egs/steps/cleanup/internal/modify_ctm_edits.py function ReadNonScoredWords (line 105) | def ReadNonScoredWords(non_scored_words_file): function ProcessLineForNonScoredWords (line 134) | def ProcessLineForNonScoredWords(a): function ProcessUtteranceForRepetitions (line 185) | def ProcessUtteranceForRepetitions(split_lines_of_utt): function ProcessUtterance (line 307) | def ProcessUtterance(split_lines_of_utt): function ProcessData (line 318) | def ProcessData(): function PrintNonScoredStats (line 367) | def PrintNonScoredStats(): function PrintRepetitionStats (line 400) | def PrintRepetitionStats(): FILE: egs/steps/cleanup/internal/resolve_ctm_edits_overlaps.py function get_args (line 34) | def get_args(): function read_segments (line 56) | def read_segments(segments_file): function read_ctm_edits (line 79) | def read_ctm_edits(ctm_edits_file, segments): function wer (line 127) | def wer(ctm_edit_lines): function choose_best_ctm_lines (line 142) | def choose_best_ctm_lines(first_lines, second_lines, function resolve_overlaps (line 153) | def resolve_overlaps(ctm_edits, segments): function ctm_edit_line_to_string (line 285) | def ctm_edit_line_to_string(line): function write_ctm_edits (line 292) | def write_ctm_edits(ctm_edit_lines, out_file): function run (line 298) | def run(args): function main (line 325) | def main(): FILE: egs/steps/cleanup/internal/retrieve_similar_docs.py function get_args (line 99) | def get_args(): function read_map (line 164) | def read_map(file_handle, num_values_per_key=None, function get_document_ids (line 224) | def get_document_ids(source_docs, indexes): function run (line 238) | def run(args): function main (line 344) | def main(): FILE: egs/steps/cleanup/internal/segment_ctm_edits.py function IsTainted (line 119) | def IsTainted(split_line_of_utt): function ComputeSegmentCores (line 136) | def ComputeSegmentCores(split_lines_of_utt): class Segment (line 176) | class Segment(object): method __init__ (line 177) | def __init__(self, split_lines_of_utt, start_index, end_index, debug_s... method PossiblyAddTaintedLines (line 213) | def PossiblyAddTaintedLines(self): method PossiblySplitSegment (line 253) | def PossiblySplitSegment(self): method PossiblyTruncateBoundaries (line 300) | def PossiblyTruncateBoundaries(self): method RelaxBoundaryTruncation (line 330) | def RelaxBoundaryTruncation(self): method PossiblyAddUnkPadding (line 381) | def PossiblyAddUnkPadding(self): method MergeWithSegment (line 432) | def MergeWithSegment(self, other): method StartTime (line 465) | def StartTime(self): method DebugInfo (line 474) | def DebugInfo(self): method EndTime (line 481) | def EndTime(self): method Length (line 489) | def Length(self): method IsWholeUtterance (line 492) | def IsWholeUtterance(self): method JunkProportion (line 503) | def JunkProportion(self): method PossiblyTruncateStartForJunkProportion (line 526) | def PossiblyTruncateStartForJunkProportion(self): method PossiblyTruncateEndForJunkProportion (line 567) | def PossiblyTruncateEndForJunkProportion(self): method ContainsAtLeastOneScoredNonOovWord (line 611) | def ContainsAtLeastOneScoredNonOovWord(self): method Text (line 624) | def Text(self): function AccumulateSegmentStats (line 643) | def AccumulateSegmentStats(segment_list, text): function PrintSegmentStats (line 649) | def PrintSegmentStats(): function GetSegmentsForUtterance (line 680) | def GetSegmentsForUtterance(split_lines_of_utt): function FloatToString (line 789) | def FloatToString(f): function TimeToString (line 800) | def TimeToString(time, frame_length): function WriteSegmentsForUtterance (line 810) | def WriteSegmentsForUtterance(text_output_handle, segments_output_handle, function PrintDebugInfoForUtterance (line 836) | def PrintDebugInfoForUtterance(ctm_edits_out_handle, function AccWordStatsForUtterance (line 880) | def AccWordStatsForUtterance(split_lines_of_utt, function PrintWordStats (line 896) | def PrintWordStats(word_stats_out): function ProcessData (line 926) | def ProcessData(): function ReadNonScoredWords (line 995) | def ReadNonScoredWords(non_scored_words_file): FILE: egs/steps/cleanup/internal/segment_ctm_edits_mild.py function non_scored_words (line 44) | def non_scored_words(): function get_args (line 48) | def get_args(): function is_tainted (line 245) | def is_tainted(split_line_of_utt): function compute_segment_cores (line 250) | def compute_segment_cores(split_lines_of_utt): class SegmentStats (line 315) | class SegmentStats(object): method __init__ (line 318) | def __init__(self): method wer (line 327) | def wer(self): method bad_proportion (line 334) | def bad_proportion(self): method incorrect_proportion (line 342) | def incorrect_proportion(self): method combine (line 349) | def combine(self, other, scale=1): method assert_equal (line 360) | def assert_equal(self, other): method compare (line 375) | def compare(self, other): method __str__ (line 393) | def __str__(self): class Segment (line 410) | class Segment(object): method __init__ (line 413) | def __init__(self, split_lines_of_utt, start_index, end_index, method copy (line 458) | def copy(self, copy_stats=True): method __str__ (line 469) | def __str__(self): method compute_stats (line 472) | def compute_stats(self): method possibly_add_tainted_lines (line 532) | def possibly_add_tainted_lines(self): method possibly_split_segment (line 583) | def possibly_split_segment(self, max_internal_silence_length, method possibly_split_long_segment (line 646) | def possibly_split_long_segment(self, max_segment_length, method possibly_truncate_boundaries (line 789) | def possibly_truncate_boundaries(self, max_edge_silence_length, method relax_boundary_truncation (line 826) | def relax_boundary_truncation(self, min_segment_length, method possibly_add_unk_padding (line 890) | def possibly_add_unk_padding(self, max_unk_padding): method start_time (line 940) | def start_time(self): method debug_info (line 955) | def debug_info(self, include_stats=True): method end_time (line 978) | def end_time(self): method length (line 991) | def length(self): method is_whole_utterance (line 995) | def is_whole_utterance(self): method get_junk_proportion (line 1005) | def get_junk_proportion(self): method get_junk_duration (line 1024) | def get_junk_duration(self): method merge_adjacent_segment (line 1028) | def merge_adjacent_segment(self, other): method merge_with_segment (line 1073) | def merge_with_segment(self, other, max_deleted_words): method contains_atleast_one_scored_non_oov_word (line 1163) | def contains_atleast_one_scored_non_oov_word(self): method text (line 1179) | def text(self, oov_symbol, eps_symbol=""): class SegmentsMerger (line 1195) | class SegmentsMerger(object): method __init__ (line 1209) | def __init__(self, segments): method _get_merged_cluster (line 1240) | def _get_merged_cluster(self, cluster1, cluster2, rejected_clusters=None, method merge_clusters (line 1306) | def merge_clusters(self, scoring_function, function merge_segments (line 1402) | def merge_segments(segments, args): function get_segments_for_utterance (line 1502) | def get_segments_for_utterance(split_lines_of_utt, args, utterance_stats): function float_to_string (line 1741) | def float_to_string(f): function time_to_string (line 1754) | def time_to_string(time, frame_length): function write_segments_for_utterance (line 1768) | def write_segments_for_utterance(text_output_handle, segments_output_han... function print_debug_info_for_utterance (line 1794) | def print_debug_info_for_utterance(ctm_edits_out_handle, class WordStats (line 1840) | class WordStats(object): method __init__ (line 1848) | def __init__(self): method accumulate_for_utterance (line 1851) | def accumulate_for_utterance(self, split_lines_of_utt, method print (line 1867) | def print(self, word_stats_out): function process_data (line 1896) | def process_data(args, oov_symbol, utterance_stats, word_stats): function read_non_scored_words (line 1951) | def read_non_scored_words(non_scored_words_file): class UtteranceStats (line 1962) | class UtteranceStats(object): method __init__ (line 1964) | def __init__(self): method accumulate_segment_stats (line 1975) | def accumulate_segment_stats(self, segment_list, text): method print_segment_stats (line 1984) | def print_segment_stats(self): function main (line 2010) | def main(): FILE: egs/steps/cleanup/internal/stitch_documents.py function get_args (line 46) | def get_args(): function run (line 82) | def run(args): function main (line 140) | def main(): FILE: egs/steps/cleanup/internal/taint_ctm_edits.py function ProcessUtterance (line 81) | def ProcessUtterance(split_lines_of_utt, remove_deletions=True): function ProcessData (line 141) | def ProcessData(): function PrintNonScoredStats (line 191) | def PrintNonScoredStats(): function PrintStats (line 220) | def PrintStats(): FILE: egs/steps/cleanup/internal/tf_idf.py class IDFStats (line 21) | class IDFStats(object): method __init__ (line 24) | def __init__(self): method get_inverse_document_frequency (line 28) | def get_inverse_document_frequency(self, term, weighting_scheme="log"): method accumulate (line 57) | def accumulate(self, term): method write (line 64) | def write(self, file_handle): method read (line 76) | def read(self, file_handle): class TFStats (line 89) | class TFStats(object): method __init__ (line 93) | def __init__(self): method get_term_frequency (line 97) | def get_term_frequency(self, term, doc, weighting_scheme="raw", method accumulate (line 124) | def accumulate(self, doc, text, ngram_order): method compute_term_stats (line 133) | def compute_term_stats(self, idf_stats=None): method __str__ (line 147) | def __str__(self): method read (line 159) | def read(self, file_handle, ngram_order=None, idf_stats=None): class TFIDF (line 187) | class TFIDF(object): method __init__ (line 195) | def __init__(self): method get_value (line 198) | def get_value(self, term, doc): method compute_similarity_scores (line 204) | def compute_similarity_scores(self, source_tfidf, source_docs=None, method read (line 273) | def read(self, tf_idf_file): method write (line 327) | def write(self, tf_idf_file): function write_tfidf_from_stats (line 340) | def write_tfidf_from_stats( function read_key (line 392) | def read_key(fd): function read_tfidf_ark (line 407) | def read_tfidf_ark(file_handle): FILE: egs/steps/cleanup/make_biased_lms.py function ProcessGroupOfLines (line 47) | def ProcessGroupOfLines(group_of_lines): FILE: egs/steps/data/augment_data_dir.py function get_args (line 21) | def get_args(): function check_args (line 71) | def check_args(args): function get_noise_list (line 93) | def get_noise_list(noise_wav_scp_filename): function augment_wav (line 104) | def augment_wav(utt, wav, dur, fg_snr_opts, bg_snr_opts, fg_noise_utts, \ function get_new_id (line 153) | def get_new_id(utt, utt_modifier_type, utt_modifier): function copy_file_if_exists (line 167) | def copy_file_if_exists(input_file, output_file, utt_modifier_type, function create_augmented_utt2uniq (line 186) | def create_augmented_utt2uniq(input_dir, output_dir, function main (line 197) | def main(): FILE: egs/steps/data/data_dir_manipulation_lib.py function RunKaldiCommand (line 3) | def RunKaldiCommand(command, wait = True): FILE: egs/steps/data/make_musan.py function get_args (line 13) | def get_args(): function check_args (line 34) | def check_args(args): function process_music_annotations (line 43) | def process_music_annotations(path): function prepare_music (line 54) | def prepare_music(root_dir, use_vocals, sampling_rate): function prepare_speech (line 92) | def prepare_speech(root_dir, sampling_rate): function prepare_noise (line 125) | def prepare_noise(root_dir, sampling_rate): function main (line 158) | def main(): FILE: egs/steps/data/reverberate_data_dir.py function get_args (line 12) | def get_args(): function check_args (line 86) | def check_args(args): class list_cyclic_iterator (line 119) | class list_cyclic_iterator(object): method __init__ (line 120) | def __init__(self, list): method __next__ (line 125) | def __next__(self): function pick_item_with_probability (line 132) | def pick_item_with_probability(x): function parse_file_to_dict (line 155) | def parse_file_to_dict(file, assert2fields = False, value_processor = No... function write_dict_to_file (line 170) | def write_dict_to_file(dict, file_name): function create_corrupted_utt2uniq (line 186) | def create_corrupted_utt2uniq(input_dir, output_dir, num_replicas, inclu... function add_point_source_noise (line 206) | def add_point_source_noise(noise_addition_descriptor, # descriptor to s... function generate_reverberation_opts (line 240) | def generate_reverberation_opts(room_dict, # the room dictionary, pleas... function get_new_id (line 302) | def get_new_id(id, prefix=None, copy=0): function generate_reverberated_wav_scp (line 315) | def generate_reverberated_wav_scp(wav_scp, # a dictionary whose values ... function add_prefix_to_fields (line 379) | def add_prefix_to_fields(input_file, output_file, num_replicas, include_... function create_reverberated_copy (line 401) | def create_reverberated_copy(input_dir, function smooth_probability_distribution (line 458) | def smooth_probability_distribution(set_list, smoothing_weight=0.0, targ... function parse_set_parameter_strings (line 493) | def parse_set_parameter_strings(set_para_array): function parse_rir_list (line 516) | def parse_rir_list(rir_set_para_array, smoothing_weight, sampling_rate =... function almost_equal (line 552) | def almost_equal(value_1, value_2, accuracy = 10**-8): function make_room_dict (line 558) | def make_room_dict(rir_list): function parse_noise_list (line 581) | def parse_noise_list(noise_set_para_array, smoothing_weight, sampling_ra... function main (line 641) | def main(): FILE: egs/steps/diagnostic/analyze_lattice_depth_stats.py function GetPercentile (line 116) | def GetPercentile(depth_to_count, fraction): function GetMean (line 131) | def GetMean(depth_to_count): FILE: egs/steps/diagnostic/analyze_phone_length_stats.py function GetPercentile (line 145) | def GetPercentile(length_to_count, fraction): function GetMean (line 160) | def GetMean(length_to_count): FILE: egs/steps/dict/apply_lexicon_edits.py function GetArgs (line 10) | def GetArgs(): function CheckArgs (line 37) | def CheckArgs(args): function ReadLexicon (line 51) | def ReadLexicon(lexicon_file_handle): function ApplyLexiconEdits (line 66) | def ApplyLexiconEdits(lexicon, lexicon_edits_file_handle): function WriteLexicon (line 98) | def WriteLexicon(lexicon, out_lexicon_handle): function Main (line 103) | def Main(): FILE: egs/steps/dict/get_pron_stats.py function GetArgs (line 12) | def GetArgs(): function CheckArgs (line 41) | def CheckArgs(args): function GetStatsFromArcInfo (line 56) | def GetStatsFromArcInfo(arc_info_file_handle, phone_map_handle): function WriteStats (line 85) | def WriteStats(stats, file_handle): function Main (line 91) | def Main(): FILE: egs/steps/dict/internal/get_subsegments.py function GetArgs (line 12) | def GetArgs(): function CheckArgs (line 43) | def CheckArgs(args): function GetSubsegments (line 60) | def GetSubsegments(args, vocab): function ReadVocab (line 119) | def ReadVocab(vocab_file_handle): function Main (line 133) | def Main(): FILE: egs/steps/dict/internal/prune_pron_candidates.py function GetArgs (line 12) | def GetArgs(): function CheckArgs (line 51) | def CheckArgs(args): function ReadStats (line 61) | def ReadStats(pron_stats_handle): function ReadLexicon (line 77) | def ReadLexicon(lexicon_handle): function ReadLexiconp (line 91) | def ReadLexiconp(lexiconp_handle): function PruneProns (line 108) | def PruneProns(args, stats, ref_lexicon, lexicon_phonetic_decoding, lexi... function Main (line 148) | def Main(): FILE: egs/steps/dict/internal/sum_arc_info.py class StrToBoolAction (line 11) | class StrToBoolAction(argparse.Action): method __call__ (line 14) | def __call__(self, parser, namespace, values, option_string=None): function GetArgs (line 23) | def GetArgs(): function CheckArgs (line 52) | def CheckArgs(args): function Main (line 67) | def Main(): FILE: egs/steps/dict/merge_learned_lexicons.py function GetArgs (line 12) | def GetArgs(): function CheckArgs (line 62) | def CheckArgs(args): function ReadArcStats (line 77) | def ReadArcStats(arc_stats_file_handle): function ReadWordCounts (line 99) | def ReadWordCounts(word_counts_file_handle): function ReadLexicon (line 111) | def ReadLexicon(args, lexicon_file_handle, counts): function WriteEditsAndSummary (line 129) | def WriteEditsAndSummary(args, learned_lexicon, ref_lexicon, pd_lexicon,... function WriteLearnedLexiconOov (line 237) | def WriteLearnedLexiconOov(learned_lexicon, ref_lexicon, file_handle): function Main (line 244) | def Main(): FILE: egs/steps/dict/prons_to_lexicon.py class StrToBoolAction (line 13) | class StrToBoolAction(argparse.Action): method __call__ (line 16) | def __call__(self, parser, namespace, values, option_string=None): function GetArgs (line 24) | def GetArgs(): function CheckArgs (line 64) | def CheckArgs(args): function ReadStats (line 87) | def ReadStats(args): function ReadLexicon (line 104) | def ReadLexicon(lexicon_file_handle): function ConvertWordCountsToProbs (line 119) | def ConvertWordCountsToProbs(args, lexicon, word_count): function ConvertWordProbsToLexicon (line 132) | def ConvertWordProbsToLexicon(word_probs): function NormalizeLexicon (line 139) | def NormalizeLexicon(lexicon, set_max_to_one = True, function TakeTopN (line 155) | def TakeTopN(lexicon, top_N): function WriteLexicon (line 167) | def WriteLexicon(args, lexicon, filter_lexicon): function Main (line 188) | def Main(): FILE: egs/steps/dict/prune_pron_candidates.py function GetArgs (line 13) | def GetArgs(): function CheckArgs (line 42) | def CheckArgs(args): function ReadStats (line 51) | def ReadStats(pron_stats_handle): function ReadLexicon (line 69) | def ReadLexicon(ref_lexicon_handle): function PruneProns (line 86) | def PruneProns(args, stats, ref_lexicon): function Main (line 114) | def Main(): FILE: egs/steps/dict/select_prons_bayesian.py function GetArgs (line 13) | def GetArgs(): function CheckArgs (line 91) | def CheckArgs(args): function ReadPronStats (line 115) | def ReadPronStats(pron_stats_file_handle): function ReadWordCounts (line 130) | def ReadWordCounts(word_counts_file_handle): function ReadLexicon (line 142) | def ReadLexicon(args, lexicon_file_handle, counts): function FilterPhoneticDecodingLexicon (line 160) | def FilterPhoneticDecodingLexicon(args, phonetic_decoding_lexicon, stats): function ComputePriorCounts (line 184) | def ComputePriorCounts(args, counts, ref_lexicon, g2p_lexicon, phonetic_... function ComputePosteriors (line 204) | def ComputePosteriors(args, stats, ref_lexicon, g2p_lexicon, phonetic_de... function SelectPronsBayesian (line 258) | def SelectPronsBayesian(args, counts, posteriors, ref_lexicon, g2p_lexic... function WriteEditsAndSummary (line 316) | def WriteEditsAndSummary(args, learned_lexicon, ref_lexicon, phonetic_de... function WriteLearnedLexiconOov (line 414) | def WriteLearnedLexiconOov(learned_lexicon, ref_lexicon, file_handle): function Main (line 421) | def Main(): FILE: egs/steps/dict/select_prons_greedy.py function GetArgs (line 12) | def GetArgs(): function CheckArgs (line 69) | def CheckArgs(args): function ReadArcStats (line 118) | def ReadArcStats(arc_stats_file_handle): function ReadWordCounts (line 140) | def ReadWordCounts(word_counts_file_handle): function ReadLexicon (line 152) | def ReadLexicon(args, lexicon_file_handle, counts): function FilterPhoneticDecodingLexicon (line 170) | def FilterPhoneticDecodingLexicon(args, pd_lexicon): function OneEMIter (line 187) | def OneEMIter(args, word, stats, prons, pron_probs, debug=False): function SelectPronsGreedy (line 212) | def SelectPronsGreedy(args, stats, counts, ref_lexicon, g2p_lexicon, pd_... function WriteLearnedLexicon (line 352) | def WriteLearnedLexicon(learned_lexicon, file_handle): function Main (line 358) | def Main(): FILE: egs/steps/libs/common.py function send_mail (line 31) | def send_mail(message, subject, email_id): function str_to_bool (line 44) | def str_to_bool(value): class StrToBoolAction (line 53) | class StrToBoolAction(argparse.Action): method __call__ (line 57) | def __call__(self, parser, namespace, values, option_string=None): class NullstrToNoneAction (line 65) | class NullstrToNoneAction(argparse.Action): method __call__ (line 70) | def __call__(self, parser, namespace, values, option_string=None): class smart_open (line 77) | class smart_open(object): method __init__ (line 87) | def __init__(self, filename, mode="r"): method __enter__ (line 92) | def __enter__(self): method __exit__ (line 101) | def __exit__(self, *args): function check_if_cuda_compiled (line 106) | def check_if_cuda_compiled(): function execute_command (line 115) | def execute_command(command): function get_command_stdout (line 132) | def get_command_stdout(command, require_zero_status = True): function wait_for_background_commands (line 159) | def wait_for_background_commands(): function background_command (line 168) | def background_command(command, require_zero_status = False): function background_command_waiter (line 198) | def background_command_waiter(command, popen_object, require_zero_status): function get_number_of_leaves_from_tree (line 215) | def get_number_of_leaves_from_tree(alidir): function get_number_of_leaves_from_model (line 226) | def get_number_of_leaves_from_model(dir): function get_number_of_jobs (line 238) | def get_number_of_jobs(alidir): function get_ivector_dim (line 248) | def get_ivector_dim(ivector_dir=None): function get_ivector_extractor_id (line 257) | def get_ivector_extractor_id(ivector_dir=None): function get_feat_dim (line 268) | def get_feat_dim(feat_dir): function get_feat_dim_from_scp (line 278) | def get_feat_dim_from_scp(feat_scp): function read_kaldi_matrix (line 286) | def read_kaldi_matrix(matrix_file): function write_kaldi_matrix (line 308) | def write_kaldi_matrix(output_file, matrix): function write_matrix_ascii (line 329) | def write_matrix_ascii(file_or_fd, mat, key=None): function read_matrix_ascii (line 365) | def read_matrix_ascii(file_or_fd): function read_key (line 403) | def read_key(fd): function read_mat_ark (line 421) | def read_mat_ark(file_or_fd): function force_symlink (line 448) | def force_symlink(file1, file2): function compute_lifter_coeffs (line 458) | def compute_lifter_coeffs(lifter, dim): function compute_idct_matrix (line 466) | def compute_idct_matrix(K, N, cepstral_lifter=0): function write_idct_matrix (line 488) | def write_idct_matrix(feat_dim, cepstral_lifter, file_path): FILE: egs/steps/libs/nnet3/report/log_parse.py class KaldiLogParseException (line 48) | class KaldiLogParseException(Exception): method __init__ (line 52) | def __init__(self, message = None): function fill_nonlin_stats_table_with_regex_result (line 63) | def fill_nonlin_stats_table_with_regex_result(groups, gate_index, stats_... function parse_progress_logs_for_nonlinearity_stats (line 148) | def parse_progress_logs_for_nonlinearity_stats(exp_dir): function parse_difference_string (line 201) | def parse_difference_string(string): class MalformedClippedProportionLineException (line 209) | class MalformedClippedProportionLineException(Exception): method __init__ (line 210) | def __init__(self, line): function parse_progress_logs_for_clipped_proportion (line 216) | def parse_progress_logs_for_clipped_proportion(exp_dir): function parse_progress_logs_for_param_diff (line 292) | def parse_progress_logs_for_param_diff(exp_dir, pattern): function get_train_times (line 366) | def get_train_times(exp_dir): function parse_prob_logs (line 390) | def parse_prob_logs(exp_dir, key='accuracy', output="output"): function parse_rnnlm_prob_logs (line 447) | def parse_rnnlm_prob_logs(exp_dir, key='objf'): function generate_acc_logprob_report (line 512) | def generate_acc_logprob_report(exp_dir, key="accuracy", output="output"): FILE: egs/steps/libs/nnet3/train/chain_objf/acoustic_model.py function create_phone_lm (line 25) | def create_phone_lm(dir, tree_dir, run_opts, lm_opts=None): function create_denominator_fst (line 53) | def create_denominator_fst(dir, tree_dir, run_opts): function generate_chain_egs (line 65) | def generate_chain_egs(dir, data, lat_dir, egs_dir, function train_new_models (line 121) | def train_new_models(dir, iter, srand, num_jobs, function train_one_iteration (line 238) | def train_one_iteration(dir, iter, srand, egs_dir, function check_for_required_files (line 369) | def check_for_required_files(feat_dir, tree_dir, lat_dir=None): function compute_preconditioning_matrix (line 381) | def compute_preconditioning_matrix(dir, egs_dir, num_lda_jobs, run_opts, function prepare_initial_acoustic_model (line 446) | def prepare_initial_acoustic_model(dir, run_opts, srand=-1, input_model=... function compute_train_cv_probabilities (line 472) | def compute_train_cv_probabilities(dir, iter, egs_dir, l2_regularize, function compute_progress (line 520) | def compute_progress(dir, iter, run_opts): function combine_models (line 558) | def combine_models(dir, num_iters, models_to_combine, num_chunk_per_mini... FILE: egs/steps/libs/nnet3/train/common.py class RunOpts (line 27) | class RunOpts(object): method __init__ (line 35) | def __init__(self): function get_outputs_list (line 44) | def get_outputs_list(model_file, get_raw_nnet_from_am=True): function get_multitask_egs_opts (line 60) | def get_multitask_egs_opts(egs_dir, egs_prefix="", function get_successful_models (line 107) | def get_successful_models(num_models, log_file_pattern, function get_average_nnet_model (line 143) | def get_average_nnet_model(dir, iter, nnets_list, run_opts, function get_best_nnet_model (line 166) | def get_best_nnet_model(dir, iter, best_model_index, run_opts, function validate_chunk_width (line 191) | def validate_chunk_width(chunk_width): function principal_chunk_width (line 209) | def principal_chunk_width(chunk_width): function validate_range_str (line 217) | def validate_range_str(range_str): function validate_minibatch_size_str (line 244) | def validate_minibatch_size_str(minibatch_size_str): function halve_range_str (line 282) | def halve_range_str(range_str): function halve_minibatch_size_str (line 299) | def halve_minibatch_size_str(minibatch_size_str): function copy_egs_properties_to_exp_dir (line 321) | def copy_egs_properties_to_exp_dir(egs_dir, dir): function parse_generic_config_vars_file (line 334) | def parse_generic_config_vars_file(var_file): function get_input_model_info (line 364) | def get_input_model_info(input_model): function verify_egs_dir (line 394) | def verify_egs_dir(egs_dir, feat_dim, ivector_dim, ivector_extractor_id, function compute_presoftmax_prior_scale (line 495) | def compute_presoftmax_prior_scale(dir, alidir, num_jobs, run_opts, function smooth_presoftmax_prior_scale_vector (line 528) | def smooth_presoftmax_prior_scale_vector(pdf_counts, function prepare_initial_network (line 542) | def prepare_initial_network(dir, run_opts, srand=-3, input_model=None): function get_model_combine_iters (line 561) | def get_model_combine_iters(num_iters, num_epochs, function get_current_num_jobs (line 605) | def get_current_num_jobs(it, num_it, start, step, end): function get_learning_rate (line 615) | def get_learning_rate(iter, num_jobs, num_iters, num_archives_processed, function should_do_shrinkage (line 630) | def should_do_shrinkage(iter, model_file, shrink_saturation_threshold, function remove_nnet_egs (line 657) | def remove_nnet_egs(egs_dir): function clean_nnet_dir (line 662) | def clean_nnet_dir(nnet_dir, num_iters, egs_dir, function remove_model (line 679) | def remove_model(nnet_dir, iter, num_iters, models_to_combine=None, function positive_int (line 695) | def positive_int(arg): class CommonParser (line 702) | class CommonParser(object): method __init__ (line 713) | def __init__(self, class SelfTest (line 1000) | class SelfTest(unittest.TestCase): method test_halve_minibatch_size_str (line 1002) | def test_halve_minibatch_size_str(self): method test_validate_chunk_width (line 1009) | def test_validate_chunk_width(self): method test_validate_minibatch_size_str (line 1014) | def test_validate_minibatch_size_str(self): method test_get_current_num_jobs (line 1026) | def test_get_current_num_jobs(self): FILE: egs/steps/libs/nnet3/train/dropout_schedule.py function _parse_dropout_option (line 18) | def _parse_dropout_option(dropout_option): function _parse_dropout_string (line 68) | def _parse_dropout_string(dropout_str): function _get_component_dropout (line 131) | def _get_component_dropout(dropout_schedule, data_fraction): function _get_dropout_proportions (line 187) | def _get_dropout_proportions(dropout_schedule, data_fraction): function get_dropout_edit_option (line 226) | def get_dropout_edit_option(dropout_schedule, data_fraction, iter_): function get_dropout_edit_string (line 269) | def get_dropout_edit_string(dropout_schedule, data_fraction, iter_): function _self_test (line 308) | def _self_test(): FILE: egs/steps/libs/nnet3/train/frame_level_objf/acoustic_model.py function generate_egs (line 21) | def generate_egs(data, alidir, egs_dir, function prepare_initial_acoustic_model (line 64) | def prepare_initial_acoustic_model(dir, alidir, run_opts, FILE: egs/steps/libs/nnet3/train/frame_level_objf/common.py function train_new_models (line 28) | def train_new_models(dir, iter, srand, num_jobs, function train_one_iteration (line 174) | def train_one_iteration(dir, iter, srand, egs_dir, function compute_preconditioning_matrix (line 320) | def compute_preconditioning_matrix(dir, egs_dir, num_lda_jobs, run_opts, function compute_train_cv_probabilities (line 380) | def compute_train_cv_probabilities(dir, iter, egs_dir, run_opts, function compute_progress (line 438) | def compute_progress(dir, iter, egs_dir, function combine_models (line 476) | def combine_models(dir, num_iters, models_to_combine, egs_dir, function get_realign_iters (line 562) | def get_realign_iters(realign_times, num_iters, function align (line 589) | def align(dir, data, lang, run_opts, iter=None, function realign (line 620) | def realign(dir, iter, feat_dir, lang, prev_egs_dir, cur_egs_dir, function adjust_am_priors (line 653) | def adjust_am_priors(dir, input_model, avg_posterior_vector, output_model, function compute_average_posterior (line 665) | def compute_average_posterior(dir, iter, egs_dir, num_archives, FILE: egs/steps/libs/nnet3/train/frame_level_objf/raw_model.py function generate_egs_using_targets (line 20) | def generate_egs_using_targets(data, targets_scp, egs_dir, FILE: egs/steps/libs/nnet3/xconfig/attention.py class XconfigAttentionLayer (line 27) | class XconfigAttentionLayer(XconfigLayerBase): method __init__ (line 28) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 37) | def set_default_configs(self): method check_configs (line 62) | def check_configs(self): method output_name (line 80) | def output_name(self, auxiliary_output=None): method attention_input_dim (line 91) | def attention_input_dim(self): method attention_output_dim (line 100) | def attention_output_dim(self): method output_dim (line 109) | def output_dim(self, auxiliary_output = None): method get_full_config (line 112) | def get_full_config(self): method _generate_config (line 124) | def _generate_config(self): method _add_components (line 139) | def _add_components(self, input_desc, input_dim, nonlinearities): FILE: egs/steps/libs/nnet3/xconfig/basic_layers.py class XconfigLayerBase (line 20) | class XconfigLayerBase(object): method __init__ (line 24) | def __init__(self, first_token, key_to_value, all_layers): method set_configs (line 79) | def set_configs(self, key_to_value, all_layers): method str (line 145) | def str(self): method __str__ (line 168) | def __str__(self): method normalize_descriptors (line 171) | def normalize_descriptors(self): method convert_to_descriptor (line 183) | def convert_to_descriptor(self, descriptor_string, all_layers): method get_dim_for_descriptor (line 203) | def get_dim_for_descriptor(self, descriptor, all_layers): method get_string_for_descriptor (line 213) | def get_string_for_descriptor(self, descriptor, all_layers): method get_name (line 224) | def get_name(self): method set_default_configs (line 233) | def set_default_configs(self): method set_derived_configs (line 239) | def set_derived_configs(self): method check_configs (line 246) | def check_configs(self): method get_input_descriptor_names (line 252) | def get_input_descriptor_names(self): method auxiliary_outputs (line 268) | def auxiliary_outputs(self): method output_name (line 278) | def output_name(self, auxiliary_output=None): method output_dim (line 293) | def output_dim(self, auxiliary_output=None): method get_full_config (line 300) | def get_full_config(self): class XconfigInputLayer (line 314) | class XconfigInputLayer(XconfigLayerBase): method __init__ (line 321) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 326) | def set_default_configs(self): method check_configs (line 330) | def check_configs(self): method get_input_descriptor_names (line 336) | def get_input_descriptor_names(self): method output_name (line 340) | def output_name(self, auxiliary_outputs=None): method output_dim (line 346) | def output_dim(self, auxiliary_outputs=None): method get_full_config (line 352) | def get_full_config(self): class XconfigTrivialOutputLayer (line 364) | class XconfigTrivialOutputLayer(XconfigLayerBase): method __init__ (line 381) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 386) | def set_default_configs(self): method check_configs (line 394) | def check_configs(self): method output_name (line 402) | def output_name(self, auxiliary_outputs=None): method output_dim (line 409) | def output_dim(self, auxiliary_outputs=None): method get_full_config (line 415) | def get_full_config(self): class XconfigOutputLayer (line 446) | class XconfigOutputLayer(XconfigLayerBase): method __init__ (line 485) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 490) | def set_default_configs(self): method check_configs (line 522) | def check_configs(self): method auxiliary_outputs (line 539) | def auxiliary_outputs(self): method output_name (line 547) | def output_name(self, auxiliary_output=None): method output_dim (line 562) | def output_dim(self, auxiliary_output=None): method get_full_config (line 572) | def get_full_config(self): method _generate_config (line 584) | def _generate_config(self): class XconfigBasicLayer (line 672) | class XconfigBasicLayer(XconfigLayerBase): method __init__ (line 706) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 709) | def set_default_configs(self): method check_configs (line 739) | def check_configs(self): method output_name (line 757) | def output_name(self, auxiliary_output=None): method output_dim (line 768) | def output_dim(self, auxiliary_output=None): method get_full_config (line 776) | def get_full_config(self): method _generate_config (line 787) | def _generate_config(self): method _add_components (line 802) | def _add_components(self, input_desc, input_dim, nonlinearities): class XconfigFixedAffineLayer (line 937) | class XconfigFixedAffineLayer(XconfigLayerBase): method __init__ (line 955) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 959) | def set_default_configs(self): method check_configs (line 968) | def check_configs(self): method output_name (line 972) | def output_name(self, auxiliary_output=None): method output_dim (line 978) | def output_dim(self, auxiliary_output=None): method get_full_config (line 985) | def get_full_config(self): class XconfigAffineLayer (line 1028) | class XconfigAffineLayer(XconfigLayerBase): method __init__ (line 1049) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 1053) | def set_default_configs(self): method set_derived_configs (line 1070) | def set_derived_configs(self): method check_configs (line 1075) | def check_configs(self): method output_name (line 1079) | def output_name(self, auxiliary_output=None): method output_dim (line 1085) | def output_dim(self, auxiliary_output=None): method get_full_config (line 1093) | def get_full_config(self): class XconfigIdctLayer (line 1128) | class XconfigIdctLayer(XconfigLayerBase): method __init__ (line 1150) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 1154) | def set_default_configs(self): method check_configs (line 1163) | def check_configs(self): method output_name (line 1167) | def output_name(self, auxiliary_output=None): method output_dim (line 1173) | def output_dim(self, auxiliary_output=None): method get_full_config (line 1180) | def get_full_config(self): method _generate_config (line 1193) | def _generate_config(self): class XconfigExistingLayer (line 1224) | class XconfigExistingLayer(XconfigLayerBase): method __init__ (line 1243) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 1249) | def set_default_configs(self): method check_configs (line 1252) | def check_configs(self): method get_input_descriptor_names (line 1257) | def get_input_descriptor_names(self): method output_name (line 1260) | def output_name(self, auxiliary_outputs=None): method output_dim (line 1265) | def output_dim(self, auxiliary_outputs=None): method get_full_config (line 1270) | def get_full_config(self): class XconfigSpecAugmentLayer (line 1277) | class XconfigSpecAugmentLayer(XconfigLayerBase): method __init__ (line 1297) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 1300) | def set_default_configs(self): method check_configs (line 1308) | def check_configs(self): method output_name (line 1314) | def output_name(self, auxiliary_output=None): method output_dim (line 1318) | def output_dim(self, auxiliary_output=None): method get_full_config (line 1323) | def get_full_config(self): method _generate_config (line 1336) | def _generate_config(self): function test_layers (line 1362) | def test_layers(): FILE: egs/steps/libs/nnet3/xconfig/composite_layers.py class XconfigTdnnfLayer (line 68) | class XconfigTdnnfLayer(XconfigLayerBase): method __init__ (line 70) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 74) | def set_default_configs(self): method set_derived_configs (line 86) | def set_derived_configs(self): method check_configs (line 89) | def check_configs(self): method output_name (line 113) | def output_name(self, auxiliary_output=None): method output_dim (line 127) | def output_dim(self, auxiliary_output=None): method get_full_config (line 130) | def get_full_config(self): method _generate_config (line 140) | def _generate_config(self): class XconfigPrefinalLayer (line 241) | class XconfigPrefinalLayer(XconfigLayerBase): method __init__ (line 243) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 247) | def set_default_configs(self): method set_derived_configs (line 255) | def set_derived_configs(self): method check_configs (line 258) | def check_configs(self): method output_name (line 264) | def output_name(self, auxiliary_output=None): method output_dim (line 268) | def output_dim(self, auxiliary_output=None): method get_full_config (line 271) | def get_full_config(self): method _generate_config (line 281) | def _generate_config(self): FILE: egs/steps/libs/nnet3/xconfig/convolution.py class XconfigConvLayer (line 115) | class XconfigConvLayer(XconfigLayerBase): method __init__ (line 116) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 122) | def set_default_configs(self): method set_derived_configs (line 143) | def set_derived_configs(self): method check_offsets_var (line 157) | def check_offsets_var(self, str): method check_configs (line 169) | def check_configs(self): method auxiliary_outputs (line 214) | def auxiliary_outputs(self): method output_name (line 217) | def output_name(self, auxiliary_output = None): method output_dim (line 229) | def output_dim(self, auxiliary_output = None): method get_full_config (line 233) | def get_full_config(self): method _generate_cnn_config (line 245) | def _generate_cnn_config(self): class XconfigResBlock (line 416) | class XconfigResBlock(XconfigLayerBase): method __init__ (line 417) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 421) | def set_default_configs(self): method set_derived_configs (line 443) | def set_derived_configs(self): method check_configs (line 463) | def check_configs(self): method auxiliary_outputs (line 471) | def auxiliary_outputs(self): method output_name (line 474) | def output_name(self, auxiliary_output = None): method output_dim (line 492) | def output_dim(self, auxiliary_output = None): method get_full_config (line 497) | def get_full_config(self): method _generate_normal_resblock_config (line 535) | def _generate_normal_resblock_config(self): method _generate_bottleneck_resblock_config (line 629) | def _generate_bottleneck_resblock_config(self): class XconfigRes2Block (line 775) | class XconfigRes2Block(XconfigLayerBase): method __init__ (line 776) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 780) | def set_default_configs(self): method set_derived_configs (line 803) | def set_derived_configs(self): method check_configs (line 826) | def check_configs(self): method auxiliary_outputs (line 830) | def auxiliary_outputs(self): method output_name (line 833) | def output_name(self, auxiliary_output = None): method output_dim (line 837) | def output_dim(self, auxiliary_output = None): method get_full_config (line 841) | def get_full_config(self): method _generate_normal_resblock_config (line 869) | def _generate_normal_resblock_config(self): method _generate_bottleneck_resblock_config (line 1015) | def _generate_bottleneck_resblock_config(self): class ChannelAverageLayer (line 1149) | class ChannelAverageLayer(XconfigLayerBase): method __init__ (line 1150) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1154) | def set_default_configs(self): method set_derived_configs (line 1158) | def set_derived_configs(self): method check_configs (line 1161) | def check_configs(self): method auxiliary_outputs (line 1170) | def auxiliary_outputs(self): method output_name (line 1173) | def output_name(self, auxiliary_output = None): method output_dim (line 1177) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1182) | def get_full_config(self): method _generate_channel_average_config (line 1190) | def _generate_channel_average_config(self): FILE: egs/steps/libs/nnet3/xconfig/gru.py class XconfigGruLayer (line 36) | class XconfigGruLayer(XconfigLayerBase): method __init__ (line 37) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 41) | def set_default_configs(self): method set_derived_configs (line 53) | def set_derived_configs(self): method check_configs (line 57) | def check_configs(self): method output_name (line 69) | def output_name(self, auxiliary_output = None): method output_dim (line 73) | def output_dim(self, auxiliary_output = None): method get_full_config (line 76) | def get_full_config(self): method generate_gru_config (line 88) | def generate_gru_config(self): class XconfigPgruLayer (line 197) | class XconfigPgruLayer(XconfigLayerBase): method __init__ (line 198) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 202) | def set_default_configs(self): method set_derived_configs (line 217) | def set_derived_configs(self): method check_configs (line 225) | def check_configs(self): method auxiliary_outputs (line 246) | def auxiliary_outputs(self): method output_name (line 249) | def output_name(self, auxiliary_output = None): method output_dim (line 259) | def output_dim(self, auxiliary_output = None): method get_full_config (line 270) | def get_full_config(self): method generate_pgru_config (line 282) | def generate_pgru_config(self): class XconfigNormPgruLayer (line 400) | class XconfigNormPgruLayer(XconfigLayerBase): method __init__ (line 401) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 405) | def set_default_configs(self): method set_derived_configs (line 422) | def set_derived_configs(self): method check_configs (line 430) | def check_configs(self): method auxiliary_outputs (line 456) | def auxiliary_outputs(self): method output_name (line 459) | def output_name(self, auxiliary_output = None): method output_dim (line 469) | def output_dim(self, auxiliary_output = None): method get_full_config (line 480) | def get_full_config(self): method generate_pgru_config (line 492) | def generate_pgru_config(self): class XconfigOpgruLayer (line 631) | class XconfigOpgruLayer(XconfigLayerBase): method __init__ (line 632) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 636) | def set_default_configs(self): method set_derived_configs (line 651) | def set_derived_configs(self): method check_configs (line 659) | def check_configs(self): method auxiliary_outputs (line 680) | def auxiliary_outputs(self): method output_name (line 683) | def output_name(self, auxiliary_output = None): method output_dim (line 693) | def output_dim(self, auxiliary_output = None): method get_full_config (line 704) | def get_full_config(self): method generate_pgru_config (line 716) | def generate_pgru_config(self): class XconfigNormOpgruLayer (line 834) | class XconfigNormOpgruLayer(XconfigLayerBase): method __init__ (line 835) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 839) | def set_default_configs(self): method set_derived_configs (line 857) | def set_derived_configs(self): method check_configs (line 865) | def check_configs(self): method auxiliary_outputs (line 891) | def auxiliary_outputs(self): method output_name (line 894) | def output_name(self, auxiliary_output = None): method output_dim (line 904) | def output_dim(self, auxiliary_output = None): method get_full_config (line 915) | def get_full_config(self): method generate_pgru_config (line 927) | def generate_pgru_config(self): class XconfigFastGruLayer (line 1065) | class XconfigFastGruLayer(XconfigLayerBase): method __init__ (line 1066) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1070) | def set_default_configs(self): method set_derived_configs (line 1088) | def set_derived_configs(self): method check_configs (line 1092) | def check_configs(self): method output_name (line 1104) | def output_name(self, auxiliary_output = None): method output_dim (line 1108) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1111) | def get_full_config(self): method generate_gru_config (line 1123) | def generate_gru_config(self): class XconfigFastPgruLayer (line 1225) | class XconfigFastPgruLayer(XconfigLayerBase): method __init__ (line 1226) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1230) | def set_default_configs(self): method set_derived_configs (line 1251) | def set_derived_configs(self): method check_configs (line 1259) | def check_configs(self): method auxiliary_outputs (line 1280) | def auxiliary_outputs(self): method output_name (line 1283) | def output_name(self, auxiliary_output = None): method output_dim (line 1293) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1304) | def get_full_config(self): method generate_pgru_config (line 1316) | def generate_pgru_config(self): class XconfigFastNormPgruLayer (line 1436) | class XconfigFastNormPgruLayer(XconfigLayerBase): method __init__ (line 1437) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1441) | def set_default_configs(self): method set_derived_configs (line 1464) | def set_derived_configs(self): method check_configs (line 1472) | def check_configs(self): method auxiliary_outputs (line 1498) | def auxiliary_outputs(self): method output_name (line 1501) | def output_name(self, auxiliary_output = None): method output_dim (line 1511) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1522) | def get_full_config(self): method generate_pgru_config (line 1534) | def generate_pgru_config(self): class XconfigFastOpgruLayer (line 1681) | class XconfigFastOpgruLayer(XconfigLayerBase): method __init__ (line 1682) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1686) | def set_default_configs(self): method set_derived_configs (line 1707) | def set_derived_configs(self): method check_configs (line 1715) | def check_configs(self): method auxiliary_outputs (line 1736) | def auxiliary_outputs(self): method output_name (line 1739) | def output_name(self, auxiliary_output = None): method output_dim (line 1749) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1760) | def get_full_config(self): method generate_pgru_config (line 1772) | def generate_pgru_config(self): class XconfigFastNormOpgruLayer (line 1895) | class XconfigFastNormOpgruLayer(XconfigLayerBase): method __init__ (line 1896) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 1900) | def set_default_configs(self): method set_derived_configs (line 1923) | def set_derived_configs(self): method check_configs (line 1931) | def check_configs(self): method auxiliary_outputs (line 1957) | def auxiliary_outputs(self): method output_name (line 1960) | def output_name(self, auxiliary_output = None): method output_dim (line 1970) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1981) | def get_full_config(self): method generate_pgru_config (line 1993) | def generate_pgru_config(self): FILE: egs/steps/libs/nnet3/xconfig/lstm.py class XconfigLstmLayer (line 45) | class XconfigLstmLayer(XconfigLayerBase): method __init__ (line 46) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 50) | def set_default_configs(self): method set_derived_configs (line 64) | def set_derived_configs(self): method check_configs (line 68) | def check_configs(self): method auxiliary_outputs (line 80) | def auxiliary_outputs(self): method output_name (line 83) | def output_name(self, auxiliary_output = None): method output_dim (line 93) | def output_dim(self, auxiliary_output = None): method get_full_config (line 104) | def get_full_config(self): method _generate_lstm_config (line 116) | def _generate_lstm_config(self): class XconfigLstmpLayer (line 295) | class XconfigLstmpLayer(XconfigLayerBase): method __init__ (line 296) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 302) | def set_default_configs(self): method set_derived_configs (line 321) | def set_derived_configs(self): method check_configs (line 329) | def check_configs(self): method auxiliary_outputs (line 356) | def auxiliary_outputs(self): method output_name (line 359) | def output_name(self, auxiliary_output = None): method output_dim (line 370) | def output_dim(self, auxiliary_output = None): method get_full_config (line 381) | def get_full_config(self): method _generate_lstm_config (line 393) | def _generate_lstm_config(self): class XconfigFastLstmLayer (line 601) | class XconfigFastLstmLayer(XconfigLayerBase): method __init__ (line 602) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 606) | def set_default_configs(self): method set_derived_configs (line 626) | def set_derived_configs(self): method check_configs (line 630) | def check_configs(self): method auxiliary_outputs (line 639) | def auxiliary_outputs(self): method output_name (line 642) | def output_name(self, auxiliary_output = None): method output_dim (line 653) | def output_dim(self, auxiliary_output = None): method get_full_config (line 663) | def get_full_config(self): method _generate_lstm_config (line 675) | def _generate_lstm_config(self): class XconfigLstmbLayer (line 798) | class XconfigLstmbLayer(XconfigLayerBase): method __init__ (line 799) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 803) | def set_default_configs(self): method set_derived_configs (line 823) | def set_derived_configs(self): method check_configs (line 827) | def check_configs(self): method auxiliary_outputs (line 837) | def auxiliary_outputs(self): method output_name (line 840) | def output_name(self, auxiliary_output = None): method output_dim (line 844) | def output_dim(self, auxiliary_output = None): method get_full_config (line 848) | def get_full_config(self): method _generate_lstm_config (line 860) | def _generate_lstm_config(self): class XconfigFastLstmpLayer (line 994) | class XconfigFastLstmpLayer(XconfigLayerBase): method __init__ (line 995) | def __init__(self, first_token, key_to_value, prev_names = None): method set_default_configs (line 999) | def set_default_configs(self): method set_derived_configs (line 1022) | def set_derived_configs(self): method check_configs (line 1031) | def check_configs(self): method auxiliary_outputs (line 1050) | def auxiliary_outputs(self): method output_name (line 1053) | def output_name(self, auxiliary_output = None): method output_dim (line 1064) | def output_dim(self, auxiliary_output = None): method get_full_config (line 1075) | def get_full_config(self): method _generate_lstm_config (line 1087) | def _generate_lstm_config(self): FILE: egs/steps/libs/nnet3/xconfig/parser.py function xconfig_line_to_object (line 97) | def xconfig_line_to_object(config_line, prev_layers = None): function get_model_component_info (line 113) | def get_model_component_info(model_filename): function read_xconfig_file (line 183) | def read_xconfig_file(xconfig_filename, existing_layers=None): FILE: egs/steps/libs/nnet3/xconfig/stats_layer.py class XconfigStatsLayer (line 13) | class XconfigStatsLayer(XconfigLayerBase): method __init__ (line 37) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 41) | def set_default_configs(self): method set_derived_configs (line 46) | def set_derived_configs(self): method check_configs (line 81) | def check_configs(self): method _generate_config (line 92) | def _generate_config(self): method output_name (line 129) | def output_name(self, auxiliary_output=None): method output_dim (line 134) | def output_dim(self, auxiliary_outputs=None): method get_full_config (line 137) | def get_full_config(self): FILE: egs/steps/libs/nnet3/xconfig/trivial_layers.py class XconfigRenormComponent (line 17) | class XconfigRenormComponent(XconfigLayerBase): method __init__ (line 26) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 29) | def set_default_configs(self): method check_configs (line 33) | def check_configs(self): method output_name (line 36) | def output_name(self, auxiliary_output=None): method output_dim (line 40) | def output_dim(self, auxiliary_output=None): method get_full_config (line 45) | def get_full_config(self): method _generate_config (line 56) | def _generate_config(self): class XconfigBatchnormComponent (line 74) | class XconfigBatchnormComponent(XconfigLayerBase): method __init__ (line 86) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 89) | def set_default_configs(self): method check_configs (line 94) | def check_configs(self): method output_name (line 97) | def output_name(self, auxiliary_output=None): method output_dim (line 101) | def output_dim(self, auxiliary_output=None): method get_full_config (line 106) | def get_full_config(self): method _generate_config (line 119) | def _generate_config(self): class XconfigNoOpComponent (line 136) | class XconfigNoOpComponent(XconfigLayerBase): method __init__ (line 144) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 147) | def set_default_configs(self): method check_configs (line 150) | def check_configs(self): method output_name (line 153) | def output_name(self, auxiliary_output=None): method output_dim (line 157) | def output_dim(self, auxiliary_output=None): method get_full_config (line 162) | def get_full_config(self): method _generate_config (line 173) | def _generate_config(self): class XconfigDeltaLayer (line 189) | class XconfigDeltaLayer(XconfigLayerBase): method __init__ (line 199) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 202) | def set_default_configs(self): method check_configs (line 205) | def check_configs(self): method output_name (line 208) | def output_name(self, auxiliary_output=None): method output_dim (line 212) | def output_dim(self, auxiliary_output=None): method get_full_config (line 217) | def get_full_config(self): method _generate_config (line 228) | def _generate_config(self): class XconfigLinearComponent (line 260) | class XconfigLinearComponent(XconfigLayerBase): method __init__ (line 279) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 282) | def set_default_configs(self): method check_configs (line 291) | def check_configs(self): method output_name (line 295) | def output_name(self, auxiliary_output=None): method output_dim (line 299) | def output_dim(self, auxiliary_output=None): method get_full_config (line 304) | def get_full_config(self): method _generate_config (line 315) | def _generate_config(self): class XconfigCombineFeatureMapsLayer (line 339) | class XconfigCombineFeatureMapsLayer(XconfigLayerBase): method __init__ (line 356) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 359) | def set_default_configs(self): method check_configs (line 366) | def check_configs(self): method output_name (line 382) | def output_name(self, auxiliary_output=None): method output_dim (line 386) | def output_dim(self, auxiliary_output=None): method get_full_config (line 391) | def get_full_config(self): method _generate_config (line 402) | def _generate_config(self): class XconfigAffineComponent (line 435) | class XconfigAffineComponent(XconfigLayerBase): method __init__ (line 454) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 457) | def set_default_configs(self): method check_configs (line 466) | def check_configs(self): method output_name (line 470) | def output_name(self, auxiliary_output=None): method output_dim (line 474) | def output_dim(self, auxiliary_output=None): method get_full_config (line 479) | def get_full_config(self): method _generate_config (line 490) | def _generate_config(self): class XconfigPerElementScaleComponent (line 514) | class XconfigPerElementScaleComponent(XconfigLayerBase): method __init__ (line 533) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 536) | def set_default_configs(self): method check_configs (line 544) | def check_configs(self): method output_name (line 547) | def output_name(self, auxiliary_output=None): method output_dim (line 551) | def output_dim(self, auxiliary_output=None): method get_full_config (line 555) | def get_full_config(self): method _generate_config (line 566) | def _generate_config(self): class XconfigPerElementOffsetComponent (line 588) | class XconfigPerElementOffsetComponent(XconfigLayerBase): method __init__ (line 607) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 610) | def set_default_configs(self): method check_configs (line 618) | def check_configs(self): method output_name (line 621) | def output_name(self, auxiliary_output=None): method output_dim (line 625) | def output_dim(self, auxiliary_output=None): method get_full_config (line 629) | def get_full_config(self): method _generate_config (line 640) | def _generate_config(self): class XconfigDimRangeComponent (line 663) | class XconfigDimRangeComponent(XconfigLayerBase): method __init__ (line 672) | def __init__(self, first_token, key_to_value, prev_names=None): method set_default_configs (line 675) | def set_default_configs(self): method check_configs (line 680) | def check_configs(self): method output_name (line 691) | def output_name(self, auxiliary_output=None): method output_dim (line 695) | def output_dim(self, auxiliary_output=None): method get_full_config (line 702) | def get_full_config(self): method _generate_config (line 713) | def _generate_config(self): FILE: egs/steps/libs/nnet3/xconfig/utils.py function get_prev_names (line 22) | def get_prev_names(all_layers, current_layer): function split_layer_name (line 46) | def split_layer_name(full_layer_name): function get_dim_from_layer_name (line 67) | def get_dim_from_layer_name(all_layers, current_layer, full_layer_name): function get_string_from_layer_name (line 110) | def get_string_from_layer_name(all_layers, current_layer, full_layer_name): function convert_value_to_type (line 148) | def convert_value_to_type(key, dest_type, string_value): class Descriptor (line 187) | class Descriptor(object): method __init__ (line 188) | def __init__(self, method config_string (line 233) | def config_string(self, layer_to_string): method str (line 243) | def str(self): method __str__ (line 251) | def __str__(self): method dim (line 261) | def dim(self, layer_to_dim): function expect_token (line 294) | def expect_token(expected_item, seen_item, what_parsing): function is_valid_line_name (line 301) | def is_valid_line_name(name): function parse_new_descriptor (line 314) | def parse_new_descriptor(tokens, pos, prev_names): function replace_bracket_expressions_in_descriptor (line 497) | def replace_bracket_expressions_in_descriptor(descriptor_string, function tokenize_descriptor (line 538) | def tokenize_descriptor(descriptor_string, function parse_config_line (line 568) | def parse_config_line(orig_config_line): function test_library (line 650) | def test_library(): FILE: egs/steps/nnet2/make_multisplice_configs.py function get_convolution_index_set (line 13) | def get_convolution_index_set(x, y): function parse_splice_string (line 22) | def parse_splice_string(splice_string): function create_config_files (line 60) | def create_config_files(output_dir, params): FILE: egs/steps/nnet3/chain/e2e/train_e2e.py function get_args (line 38) | def get_args(): function process_args (line 180) | def process_args(args): function train (line 244) | def train(args, run_opts): function main (line 553) | def main(): FILE: egs/steps/nnet3/chain/train.py function get_args (line 38) | def get_args(): function process_args (line 203) | def process_args(args): function train (line 271) | def train(args, run_opts): function main (line 641) | def main(): FILE: egs/steps/nnet3/chain2/internal/get_train_schedule.py function get_args (line 22) | def get_args(): function get_schedules (line 91) | def get_schedules(args): function main (line 154) | def main(): FILE: egs/steps/nnet3/components.py function GetSumDescriptor (line 12) | def GetSumDescriptor(inputs): function AddInputLayer (line 30) | def AddInputLayer(config_lines, feat_dim, splice_indexes=[0], ivector_di... function AddNoOpLayer (line 49) | def AddNoOpLayer(config_lines, name, input): function AddLdaLayer (line 59) | def AddLdaLayer(config_lines, name, input, lda_file): function AddFixedAffineLayer (line 62) | def AddFixedAffineLayer(config_lines, name, input, matrix_file): function AddBlockAffineLayer (line 73) | def AddBlockAffineLayer(config_lines, name, input, output_dim, num_blocks): function AddPermuteLayer (line 84) | def AddPermuteLayer(config_lines, name, input, column_map): function AddAffineLayer (line 94) | def AddAffineLayer(config_lines, name, input, output_dim, ng_affine_opti... function AddAffRelNormLayer (line 107) | def AddAffRelNormLayer(config_lines, name, input, output_dim, ng_affine_... function AddAffPnormLayer (line 127) | def AddAffPnormLayer(config_lines, name, input, pnorm_input_dim, pnorm_o... function AddConvolutionLayer (line 142) | def AddConvolutionLayer(config_lines, name, input, function AddMaxpoolingLayer (line 180) | def AddMaxpoolingLayer(config_lines, name, input, function AddSoftmaxLayer (line 218) | def AddSoftmaxLayer(config_lines, name, input): function AddSigmoidLayer (line 229) | def AddSigmoidLayer(config_lines, name, input, self_repair_scale = None): function AddOutputLayer (line 240) | def AddOutputLayer(config_lines, input, label_delay = None, suffix=None,... function AddFinalLayer (line 252) | def AddFinalLayer(config_lines, input, output_dim, function AddLstmLayer (line 289) | def AddLstmLayer(config_lines, function AddBLstmLayer (line 440) | def AddBLstmLayer(config_lines, FILE: egs/steps/nnet3/convert_nnet2_to_nnet3.py function GetArgs (line 67) | def GetArgs(): class Nnet3Model (line 97) | class Nnet3Model(object): method __init__ (line 100) | def __init__(self): method add_component (line 112) | def add_component(self, component, pairs): method write_config (line 149) | def write_config(self, filename): method write_model (line 187) | def write_model(self, model, binary="true"): function parse_nnet2_to_nnet3 (line 206) | def parse_nnet2_to_nnet3(line_buffer): function parse_transition_model (line 238) | def parse_transition_model(line_buffer): function parse_nnet2_header (line 259) | def parse_nnet2_header(line_buffer): function parse_component (line 271) | def parse_component(line, line_buffer): function parse_standard_component (line 294) | def parse_standard_component(component, line, line_buffer): function parse_fixed_scale_component (line 301) | def parse_fixed_scale_component(component, line, line_buffer): function parse_sum_group_component (line 315) | def parse_sum_group_component(component, line, line_buffer): function parse_fixed_bias_component (line 323) | def parse_fixed_bias_component(component, line, line_buffer): function parse_splice_component (line 337) | def parse_splice_component(component, line, line_buffer): function parse_end_of_component (line 355) | def parse_end_of_component(component, line, line_buffer): function parse_affine_component (line 364) | def parse_affine_component(component, line, line_buffer): function parse_weights (line 386) | def parse_weights(line_buffer): function parse_bias (line 402) | def parse_bias(line): function parse_vector (line 408) | def parse_vector(line): function parse_priors (line 412) | def parse_priors(line, line_buffer): function token_to_string (line 423) | def token_to_string(token): function consume_token (line 432) | def consume_token(token, line): function make_splice_string (line 440) | def make_splice_string(nodename, context, const_component_dim=0): function Main (line 455) | def Main(): FILE: egs/steps/nnet3/dot/descriptor_parser.py function ParseSubsegmentsAndArguments (line 12) | def ParseSubsegmentsAndArguments(segment_endpoints, sub_segments, argume... function IdentifyNestedSegments (line 40) | def IdentifyNestedSegments(input_string): FILE: egs/steps/nnet3/dot/nnet3_to_dot.py function GetDotNodeName (line 89) | def GetDotNodeName(name_string, is_component = False): function ProcessAppendDescriptor (line 103) | def ProcessAppendDescriptor(segment, parent_node_name, affix, edge_attri... function ProcessRoundDescriptor (line 139) | def ProcessRoundDescriptor(segment, parent_node_name, affix, edge_attrib... function ProcessOffsetDescriptor (line 161) | def ProcessOffsetDescriptor(segment, parent_node_name, affix, edge_attri... function ProcessSumDescriptor (line 183) | def ProcessSumDescriptor(segment, parent_node_name, affix, edge_attribut... function ProcessReplaceIndexDescriptor (line 218) | def ProcessReplaceIndexDescriptor(segment, parent_node_name, affix, edge... function ProcessIfDefinedDescriptor (line 240) | def ProcessIfDefinedDescriptor(segment, parent_node_name, affix, edge_at... function DescriptorSegmentToDot (line 257) | def DescriptorSegmentToDot(segment, parent_node_name, affix, edge_attrib... function Nnet3DescriptorToDot (line 278) | def Nnet3DescriptorToDot(descriptor, parent_node_name): function ParseNnet3String (line 289) | def ParseNnet3String(string): function Nnet3ComponentToDot (line 322) | def Nnet3ComponentToDot(component_config, component_attributes = None): function Nnet3InputToDot (line 344) | def Nnet3InputToDot(parsed_config): function Nnet3OutputToDot (line 349) | def Nnet3OutputToDot(parsed_config): function Nnet3DimrangeToDot (line 356) | def Nnet3DimrangeToDot(parsed_config): function Nnet3ComponentNodeToDot (line 366) | def Nnet3ComponentNodeToDot(parsed_config): function GroupConfigs (line 375) | def GroupConfigs(configs, node_prefixes = None): function ParseConfigLines (line 395) | def ParseConfigLines(lines, node_prefixes = None, component_attributes =... FILE: egs/steps/nnet3/lstm/make_configs.py function GetArgs (line 17) | def GetArgs(): function CheckArgs (line 115) | def CheckArgs(args): function PrintConfig (line 162) | def PrintConfig(file_name, config_lines): function ParseSpliceString (line 169) | def ParseSpliceString(splice_indexes, label_delay=None): function ParseLstmDelayString (line 211) | def ParseLstmDelayString(lstm_delay): function MakeConfigs (line 232) | def MakeConfigs(config_dir, feat_dim, ivector_dim, num_targets, function ProcessSpliceIndexes (line 326) | def ProcessSpliceIndexes(config_dir, splice_indexes, label_delay, num_ls... function Main (line 347) | def Main(): FILE: egs/steps/nnet3/multilingual/allocate_multilingual_examples.py function get_args (line 60) | def get_args(): function read_lines (line 107) | def read_lines(file_handle, num_lines): function process_multilingual_egs (line 119) | def process_multilingual_egs(args): function main (line 222) | def main(): FILE: egs/steps/nnet3/report/convert_model.py function read_next_token (line 25) | def read_next_token(s, pos): function check_for_newline (line 51) | def check_for_newline(s, pos): function read_float (line 70) | def read_float(s, pos): function read_int (line 88) | def read_int(s, pos): function read_vector (line 106) | def read_vector(s, pos): function read_matrix (line 138) | def read_matrix(s, pos): function is_component_type (line 193) | def is_component_type(component_type): function read_generic (line 200) | def read_generic(s, pos, terminating_token, action_dict): function get_action_dict (line 248) | def get_action_dict(component_type): function get_stdout_from_command (line 293) | def get_stdout_from_command(command): function read_component (line 309) | def read_component(s, pos): function read_model (line 339) | def read_model(filename): function compute_derived_quantities (line 390) | def compute_derived_quantities(model): function compute_progress (line 417) | def compute_progress(model1, model2): function test (line 454) | def test(): FILE: egs/steps/nnet3/report/generate_plots.py function get_args (line 41) | def get_args(): class LatexReport (line 86) | class LatexReport(object): method __init__ (line 89) | def __init__(self, pdf_file): method add_figure (line 98) | def add_figure(self, figure_pdf, title): method close (line 117) | def close(self): method compile (line 121) | def compile(self): function latex_compliant_name (line 141) | def latex_compliant_name(name_string): function generate_acc_logprob_plots (line 152) | def generate_acc_logprob_plots(exp_dir, output_dir, plot, key='accuracy', function insert_a_column_legend (line 219) | def insert_a_column_legend(legend_handle, legend_label, lp, mp, hp, function plot_a_nonlin_component (line 229) | def plot_a_nonlin_component(fig, dirs, stat_tables_per_component_per_dir, function generate_nonlin_stats_plots (line 351) | def generate_nonlin_stats_plots(exp_dir, output_dir, plot, comparison_di... function generate_clipped_proportion_plots (line 482) | def generate_clipped_proportion_plots(exp_dir, output_dir, plot, function generate_parameter_diff_plots (line 579) | def generate_parameter_diff_plots(exp_dir, output_dir, plot, function generate_plots (line 710) | def generate_plots(exp_dir, output_dir, output_names, comparison_dir=None, function main (line 782) | def main(): FILE: egs/steps/nnet3/report/summarize_compute_debug_timing.py function GetArgs (line 19) | def GetArgs(): function FindOpenParanthesisPosition (line 38) | def FindOpenParanthesisPosition(string): function ExtractCommandName (line 63) | def ExtractCommandName(command_string): function Main (line 76) | def Main(): FILE: egs/steps/nnet3/tdnn/make_configs.py function GetArgs (line 21) | def GetArgs(): function CheckArgs (line 138) | def CheckArgs(args): function AddConvMaxpLayer (line 221) | def AddConvMaxpLayer(config_lines, name, input, args): function AddCnnLayers (line 241) | def AddCnnLayers(config_lines, cnn_layer, cnn_bottleneck_dim, cepstral_l... function PrintConfig (line 273) | def PrintConfig(file_name, config_lines): function ParseCnnString (line 280) | def ParseCnnString(cnn_param_string_list): function ParseSpliceString (line 301) | def ParseSpliceString(splice_indexes): function MakeConfigs (line 336) | def MakeConfigs(config_dir, splice_indexes_string, function Main (line 537) | def Main(): FILE: egs/steps/nnet3/train_dnn.py function get_args (line 39) | def get_args(): function process_args (line 109) | def process_args(args): function train (line 168) | def train(args, run_opts): function main (line 456) | def main(): FILE: egs/steps/nnet3/train_raw_dnn.py function get_args (line 38) | def get_args(): function process_args (line 127) | def process_args(args): function train (line 186) | def train(args, run_opts): function main (line 498) | def main(): FILE: egs/steps/nnet3/train_raw_rnn.py function get_args (line 38) | def get_args(): function process_args (line 167) | def process_args(args): function train (line 232) | def train(args, run_opts): function main (line 558) | def main(): FILE: egs/steps/nnet3/train_rnn.py function get_args (line 38) | def get_args(): function process_args (line 158) | def process_args(args): function train (line 223) | def train(args, run_opts): function main (line 545) | def main(): FILE: egs/steps/nnet3/xconfig_to_config.py function get_args (line 25) | def get_args(): function write_config_file (line 64) | def write_config_file(config_file_out, all_layers): function main (line 92) | def main(): FILE: egs/steps/nnet3/xconfig_to_configs.py function get_args (line 23) | def get_args(): function check_args (line 62) | def check_args(args): function backup_xconfig_file (line 68) | def backup_xconfig_file(xconfig_file, config_dir): function write_expanded_xconfig_files (line 99) | def write_expanded_xconfig_files(config_dir, all_layers): function get_config_headers (line 142) | def get_config_headers(): function write_config_files (line 176) | def write_config_files(config_dir, all_layers): function add_nnet_context_info (line 233) | def add_nnet_context_info(config_dir, nnet_edits=None, function check_model_contexts (line 268) | def check_model_contexts(config_dir, nnet_edits=None, existing_model=None): function main (line 317) | def main(): FILE: egs/steps/overlap/get_overlap_segments.py function get_args (line 21) | def get_args(): class Segment (line 32) | class Segment: method __init__ (line 35) | def __init__(self, reco_id, start_time, dur = None, end_time = None, s... function groupby (line 46) | def groupby(iterable, keyfunc): function find_overlapping_segments (line 52) | def find_overlapping_segments(segs, label): function find_single_speaker_segments (line 77) | def find_single_speaker_segments(segs): function main (line 108) | def main(): FILE: egs/steps/overlap/get_overlap_targets.py function get_args (line 29) | def get_args(): class Segment (line 64) | class Segment: method __init__ (line 72) | def __init__(self, reco_id, start_time, dur = None, end_time = None, l... function groupby (line 83) | def groupby(iterable, keyfunc): function run (line 89) | def run(args): function main (line 149) | def main(): FILE: egs/steps/overlap/output_to_rttm.py function get_args (line 34) | def get_args(): function to_str (line 94) | def to_str(segment): class SegmenterStats (line 100) | class SegmenterStats(object): method __init__ (line 103) | def __init__(self): method add (line 113) | def add(self, other): method __str__ (line 124) | def __str__(self): function process_label (line 143) | def process_label(text_label): class Segmentation (line 159) | class Segmentation(object): method __init__ (line 164) | def __init__(self, region_type): method initialize_segments (line 170) | def initialize_segments(self, alignment, frame_shift=0.01): method filter_short_segments (line 204) | def filter_short_segments(self, min_dur): method pad_segments (line 222) | def pad_segments(self, segment_padding, max_duration=float("inf")): method merge_consecutive_segments (line 260) | def merge_consecutive_segments(self, max_dur): method write (line 282) | def write(self, key, file_handle): function run (line 293) | def run(args): function main (line 330) | def main(): FILE: egs/steps/overlap/prepare_overlap_graph.py function get_args (line 24) | def get_args(): function print_states (line 79) | def print_states(args, file_handle): function main (line 197) | def main(): FILE: egs/steps/pytorchnn/compute_sentence_scores.py function load_nbest (line 19) | def load_nbest(path): function read_vocab (line 53) | def read_vocab(path): function get_input_and_target (line 77) | def get_input_and_target(hyp, vocab): function compute_sentence_score (line 110) | def compute_sentence_score(model, criterion, ntokens, data, target, function compute_scores (line 148) | def compute_scores(nbest, model, criterion, ntokens, vocab, model_type='... function write_scores (line 202) | def write_scores(nbest_and_scores, path): function main (line 224) | def main(): FILE: egs/steps/pytorchnn/data.py class Dictionary (line 9) | class Dictionary(object): method __init__ (line 10) | def __init__(self): method read_vocab (line 14) | def read_vocab(self, path): method __len__ (line 24) | def __len__(self): class Corpus (line 28) | class Corpus(object): method __init__ (line 29) | def __init__(self, path): method tokenize (line 36) | def tokenize(self, path): FILE: egs/steps/pytorchnn/model.py class RNNModel (line 10) | class RNNModel(nn.Module): method __init__ (line 12) | def __init__(self, rnn_type, ntoken, ninp, nhid, nlayers, dropout=0.5, method init_weights (line 41) | def init_weights(self): method forward (line 47) | def forward(self, x, hidden): method init_hidden (line 54) | def init_hidden(self, bsz): class PositionalEncoding (line 62) | class PositionalEncoding(nn.Module): method __init__ (line 79) | def __init__(self, d_model, dropout=0.1, max_len=5000): method forward (line 91) | def forward(self, x): class TransformerModel (line 106) | class TransformerModel(nn.Module): method __init__ (line 109) | def __init__(self, ntoken, ninp, nhead, nhid, nlayers, dropout=0.5, method _generate_square_subsequent_mask (line 133) | def _generate_square_subsequent_mask(self, sz): method init_weights (line 139) | def init_weights(self): method forward (line 145) | def forward(self, src, has_mask=True): FILE: egs/steps/pytorchnn/train.py function batchify (line 92) | def batchify(data, bsz, random_start_idx=False): function repackage_hidden (line 135) | def repackage_hidden(h): function get_batch (line 143) | def get_batch(source, i): function train (line 150) | def train(): function evaluate (line 189) | def evaluate(source): FILE: egs/steps/segmentation/convert_utt2spk_and_segments_to_rttm.py function get_args (line 33) | def get_args(): function main (line 55) | def main(): FILE: egs/steps/segmentation/internal/arc_info_to_targets.py function get_args (line 31) | def get_args(): function run (line 63) | def run(args): function main (line 151) | def main(): FILE: egs/steps/segmentation/internal/find_oov_phone.py function main (line 16) | def main(): FILE: egs/steps/segmentation/internal/get_default_targets_for_out_of_segments.py function get_args (line 38) | def get_args(): function run (line 90) | def run(args): function main (line 171) | def main(): FILE: egs/steps/segmentation/internal/get_transform_probs_mat.py function get_args (line 12) | def get_args(): function run (line 50) | def run(args): function main (line 73) | def main(): FILE: egs/steps/segmentation/internal/merge_segment_targets_to_recording.py function get_args (line 33) | def get_args(): function read_reco2utt_file (line 80) | def read_reco2utt_file(reco2utt_file): function read_reco2num_frames_file (line 93) | def read_reco2num_frames_file(reco2num_frames_file): function read_segments_file (line 107) | def read_segments_file(segments_file, reco2utt): function read_targets_scp (line 126) | def read_targets_scp(targets_scp, segments): function run (line 142) | def run(args): function main (line 282) | def main(): FILE: egs/steps/segmentation/internal/merge_targets.py function get_args (line 37) | def get_args(): function should_remove_frame (line 86) | def should_remove_frame(row, dim): function run (line 160) | def run(args): function main (line 205) | def main(): FILE: egs/steps/segmentation/internal/prepare_sad_graph.py function get_args (line 34) | def get_args(): function print_states (line 81) | def print_states(args, file_handle): function main (line 154) | def main(): FILE: egs/steps/segmentation/internal/resample_targets.py function get_args (line 35) | def get_args(): function run (line 70) | def run(args): function main (line 108) | def main(): FILE: egs/steps/segmentation/internal/sad_to_segments.py function get_args (line 34) | def get_args(): function to_str (line 90) | def to_str(segment): class SegmenterStats (line 96) | class SegmenterStats(object): method __init__ (line 99) | def __init__(self): method add (line 109) | def add(self, other): method __str__ (line 120) | def __str__(self): function process_label (line 139) | def process_label(text_label): class Segmentation (line 154) | class Segmentation(object): method __init__ (line 157) | def __init__(self): method initialize_segments (line 161) | def initialize_segments(self, alignment, frame_shift=0.01): method filter_short_segments (line 195) | def filter_short_segments(self, min_dur): method pad_speech_segments (line 213) | def pad_speech_segments(self, segment_padding, max_duration=float("inf... method merge_consecutive_segments (line 251) | def merge_consecutive_segments(self, max_dur): method write (line 273) | def write(self, key, file_handle): function run (line 286) | def run(args): function main (line 323) | def main(): FILE: egs/steps/segmentation/internal/verify_phones_list.py function get_args (line 12) | def get_args(): function main (line 25) | def main(): FILE: egs/steps/tfrnnlm/lstm.py class Config (line 48) | class Config(object): function data_type (line 62) | def data_type(): class RNNLMModel (line 66) | class RNNLMModel(tf.Module): method __init__ (line 69) | def __init__(self, config, logits_bias_initializer=None): method get_logits (line 97) | def get_logits(self, word_ids, is_training=False): method get_loss (line 106) | def get_loss(self, word_ids, labels, is_training=False): method get_score (line 111) | def get_score(self, logits): method get_initial_state (line 116) | def get_initial_state(self): method single_step (line 124) | def single_step(self, context, word_id): class RNNLMModelTrainer (line 142) | class RNNLMModelTrainer(tf.Module): method __init__ (line 145) | def __init__(self, model: RNNLMModel, config): method train_one_epoch (line 154) | def train_one_epoch(self, data_producer, learning_rate, verbose=True): method evaluate (line 164) | def evaluate(self, data_producer): method _train_step (line 173) | def _train_step(self, inputs, labels): function get_config (line 184) | def get_config(): function main (line 188) | def main(_): FILE: egs/steps/tfrnnlm/lstm_fast.py class Config (line 50) | class Config(object): function data_type (line 65) | def data_type(): function new_softmax (line 72) | def new_softmax(labels, logits): class MyFastLossFunction (line 87) | class MyFastLossFunction(LossFunctionWrapper): method __init__ (line 88) | def __init__(self): class FastRNNLMModel (line 92) | class FastRNNLMModel(RNNLMModel): method __init__ (line 93) | def __init__(self, config): method get_loss (line 96) | def get_loss(self, word_ids, labels, is_training=False): method get_score (line 101) | def get_score(self, logits): function get_config (line 106) | def get_config(): function main (line 110) | def main(_): FILE: egs/steps/tfrnnlm/reader.py function _read_words (line 28) | def _read_words(filename): function _build_vocab (line 32) | def _build_vocab(filename): function _file_to_word_ids (line 38) | def _file_to_word_ids(filename, word_to_id): function rnnlm_raw_data (line 43) | def rnnlm_raw_data(data_path, vocab_path): function rnnlm_gen_data (line 64) | def rnnlm_gen_data(*files): class RNNLMProducer (line 92) | class RNNLMProducer(tf.Module): method __init__ (line 95) | def __init__(self, raw_data, batch_size, num_steps, name=None): method iterate (line 112) | def iterate(self): FILE: egs/steps/tfrnnlm/vanilla_rnnlm.py class Config (line 54) | class Config(object): function data_type (line 68) | def data_type(): class RnnlmInput (line 72) | class RnnlmInput(object): method __init__ (line 75) | def __init__(self, config, data, name=None): class RnnlmModel (line 82) | class RnnlmModel(object): method __init__ (line 85) | def __init__(self, is_training, config, input_): method assign_lr (line 211) | def assign_lr(self, session, lr_value): method input (line 215) | def input(self): method initial_state (line 219) | def initial_state(self): method cost (line 223) | def cost(self): method final_state (line 227) | def final_state(self): method lr (line 231) | def lr(self): method train_op (line 235) | def train_op(self): function run_epoch (line 238) | def run_epoch(session, model, eval_op=None, verbose=False): function get_config (line 272) | def get_config(): function main (line 275) | def main(_): FILE: egs/utils/ctm/resolve_ctm_overlaps.py function get_args (line 38) | def get_args(): function read_segments (line 61) | def read_segments(segments_file): function read_ctm (line 84) | def read_ctm(ctm_file, segments): function resolve_overlaps (line 127) | def resolve_overlaps(ctms, segments): function ctm_line_to_string (line 265) | def ctm_line_to_string(line): function write_ctm (line 271) | def write_ctm(ctm_lines, out_file): function run (line 277) | def run(args): function main (line 302) | def main(): FILE: egs/utils/data/extend_segment_times.py function FloatToString (line 93) | def FloatToString(f): FILE: egs/utils/data/get_allowed_durations.py function get_args (line 32) | def get_args(): function read_kaldi_mapfile (line 89) | def read_kaldi_mapfile(path): function find_duration_range (line 104) | def find_duration_range(utt2dur, coverage_factor): function get_allowed_durations (line 136) | def get_allowed_durations(start_dur, end_dur, args): function get_trivial_allowed_durations (line 165) | def get_trivial_allowed_durations(utt2dur, args): function main (line 197) | def main(): FILE: egs/utils/data/get_uniform_subsegments.py function get_args (line 13) | def get_args(): function run (line 65) | def run(args): function main (line 109) | def main(): FILE: egs/utils/data/internal/choose_utts_to_combine.py function LessThan (line 71) | def LessThan(x, y): function CombineList (line 90) | def CombineList(min_duration, durations): function SelfTest (line 188) | def SelfTest(): function GetUtteranceGroups (line 232) | def GetUtteranceGroups(min_duration, merge_within_speakers_only, spk2utt... FILE: egs/utils/data/internal/combine_segments_to_recording.py function get_args (line 12) | def get_args(): function main (line 32) | def main(): FILE: egs/utils/data/internal/modify_speaker_info.py function SplitIntoGroups (line 65) | def SplitIntoGroups(uttlist): function GetFormatString (line 87) | def GetFormatString(d): FILE: egs/utils/data/internal/perturb_volume.py function get_args (line 18) | def get_args(): function read_reco2vol (line 50) | def read_reco2vol(volumes_file): function run (line 69) | def run(args): function main (line 109) | def main(): FILE: egs/utils/data/perturb_speed_to_allowed_lengths.py function get_args (line 30) | def get_args(): class Utterance (line 63) | class Utterance(object): method __init__ (line 68) | def __init__(self, uid, wavefile, speaker, transcription, dur): method to_kaldi_utt_str (line 76) | def to_kaldi_utt_str(self): method to_kaldi_wave_str (line 79) | def to_kaldi_wave_str(self): method to_kaldi_dur_str (line 82) | def to_kaldi_dur_str(self): function read_kaldi_datadir (line 86) | def read_kaldi_datadir(dir): function read_kaldi_mapfile (line 126) | def read_kaldi_mapfile(path): function generate_kaldi_data_files (line 140) | def generate_kaldi_data_files(utterances, outdir): function find_duration_range (line 178) | def find_duration_range(utterances, coverage_factor): function find_allowed_durations (line 212) | def find_allowed_durations(start_dur, end_dur, args): function perturb_utterances (line 242) | def perturb_utterances(utterances, allowed_durations, args): function main (line 310) | def main(): FILE: egs/utils/lang/bpe/apply_bpe.py class BPE (line 27) | class BPE(object): method __init__ (line 29) | def __init__(self, codes, merges=-1, separator='@@', vocab=None, gloss... method process_line (line 62) | def process_line(self, line): method segment (line 79) | def segment(self, sentence): method _isolate_glossaries (line 102) | def _isolate_glossaries(self, word): function create_parser (line 109) | def create_parser(): function get_pairs (line 150) | def get_pairs(word): function encode (line 162) | def encode(orig, bpe_codes, bpe_codes_reverse, vocab, separator, version... function recursive_split (line 226) | def recursive_split(segment, bpe_codes, vocab, separator, final=False): function check_vocab_and_split (line 253) | def check_vocab_and_split(orig, bpe_codes, vocab, separator): function read_vocabulary (line 278) | def read_vocabulary(vocab_file, threshold): function isolate_glossary (line 292) | def isolate_glossary(word, glossary): FILE: egs/utils/lang/bpe/bidi.py function determine_text_direction (line 15) | def determine_text_direction(text): function utf8_visual_to_logical (line 23) | def utf8_visual_to_logical(text): function utf8_logical_to_visual (line 37) | def utf8_logical_to_visual(text): FILE: egs/utils/lang/bpe/learn_bpe.py function create_parser (line 30) | def create_parser(): function get_vocabulary (line 58) | def get_vocabulary(fobj, is_dict=False): function update_pair_statistics (line 76) | def update_pair_statistics(pair, changed, stats, indices): function get_pair_statistics (line 135) | def get_pair_statistics(vocab): function replace_pair (line 154) | def replace_pair(pair, vocab, indices): function prune_stats (line 178) | def prune_stats(stats, big_stats, threshold): function main (line 194) | def main(infile, outfile, num_symbols, min_frequency=2, verbose=False, i... FILE: egs/utils/lang/compute_sentence_probs_arpa.py function check_args (line 29) | def check_args(args): function is_logprob (line 35) | def is_logprob(input): function check_number (line 45) | def check_number(model_file, tot_num): function load_model (line 58) | def load_model(model_file): function compute_sublist_prob (line 84) | def compute_sublist_prob(sub_list): function compute_begin_prob (line 97) | def compute_begin_prob(sub_list): function compute_sentence_prob (line 111) | def compute_sentence_prob(sentence, ngram_order): function output_result (line 132) | def output_result(text_in_handle, output_file_handle, ngram_order): FILE: egs/utils/lang/grammar/augment_phones_txt.py function get_args (line 9) | def get_args(): function read_phones_txt (line 27) | def read_phones_txt(filename): function read_nonterminals (line 57) | def read_nonterminals(filename): function write_phones_txt (line 73) | def write_phones_txt(orig_lines, highest_numbered_symbol, nonterminals, ... function main (line 88) | def main(): FILE: egs/utils/lang/grammar/augment_words_txt.py function get_args (line 9) | def get_args(): function read_words_txt (line 28) | def read_words_txt(filename): function read_nonterminals (line 58) | def read_nonterminals(filename): function write_words_txt (line 74) | def write_words_txt(orig_lines, highest_numbered_symbol, nonterminals, f... function main (line 88) | def main(): FILE: egs/utils/lang/internal/arpa2fst_constrained.py class HistoryState (line 48) | class HistoryState(object): method __init__ (line 49) | def __init__(self): class ArpaModel (line 60) | class ArpaModel(object): method __init__ (line 61) | def __init__(self): method Read (line 71) | def Read(self, arpa_in): method GetProb (line 167) | def GetProb(self, hist, word): method GetStateForHist (line 187) | def GetStateForHist(self, hist_to_state, hist): method GetHistToStateMap (line 199) | def GetHistToStateMap(self): method PrintAsFst (line 242) | def PrintAsFst(self, disambig_symbol, bigram_map): function ReadBigramMap (line 344) | def ReadBigramMap(bigrams_file): FILE: egs/utils/lang/limit_arpa_unk_history.py function get_ngram_stats (line 38) | def get_ngram_stats(old_lm_lines): function find_and_replace_unks (line 59) | def find_and_replace_unks(old_lm_lines, max_ngrams, skip_rows): function read_old_lm (line 122) | def read_old_lm(): function write_new_lm (line 133) | def write_new_lm(new_lm_lines, ngram_counts, ngram_diffs): function main (line 152) | def main(): FILE: egs/utils/lang/make_kn_lm.py class CountsForHistory (line 46) | class CountsForHistory: method __init__ (line 51) | def __init__(self): method words (line 60) | def words(self): method __str__ (line 63) | def __str__(self): method add_count (line 70) | def add_count(self, predicted_word, context_word, count): class NgramCounts (line 79) | class NgramCounts: method __init__ (line 88) | def __init__(self, ngram_order, bos_symbol='', eos_symbol=''): method add_count (line 105) | def add_count(self, history, predicted_word, context_word, count): method add_raw_counts_from_line (line 110) | def add_raw_counts_from_line(self, line): method add_raw_counts_from_standard_input (line 128) | def add_raw_counts_from_standard_input(self): method add_raw_counts_from_file (line 140) | def add_raw_counts_from_file(self, filename): method cal_discounting_constants (line 152) | def cal_discounting_constants(self): method cal_f (line 172) | def cal_f(self): method cal_bow (line 205) | def cal_bow(self): method print_raw_counts (line 248) | def print_raw_counts(self, info_string): method print_modified_counts (line 263) | def print_modified_counts(self, info_string): method print_f (line 284) | def print_f(self, info_string): method print_f_and_bow (line 303) | def print_f_and_bow(self, info_string): method print_as_arpa (line 326) | def print_as_arpa(self, fout=io.TextIOWrapper(sys.stdout.buffer, encod... FILE: egs/utils/lang/make_lexicon_fst.py function get_args (line 21) | def get_args(): function read_lexiconp (line 60) | def read_lexiconp(filename): function write_nonterminal_arcs (line 119) | def write_nonterminal_arcs(start_state, loop_state, next_state, function write_fst_no_silence (line 173) | def write_fst_no_silence(lexicon, nonterminals=None, left_context_phones... function write_fst_with_silence (line 220) | def write_fst_with_silence(lexicon, sil_prob, sil_phone, sil_disambig, function write_words_txt (line 308) | def write_words_txt(orig_lines, highest_numbered_symbol, nonterminals, f... function read_nonterminals (line 322) | def read_nonterminals(filename): function read_left_context_phones (line 338) | def read_left_context_phones(filename): function is_token (line 354) | def is_token(s): function main (line 363) | def main(): FILE: egs/utils/lang/make_lexicon_fst_silprob.py function get_args (line 23) | def get_args(): function read_silprobs (line 70) | def read_silprobs(filename): function read_lexiconp (line 116) | def read_lexiconp(filename): function write_nonterminal_arcs (line 184) | def write_nonterminal_arcs(start_state, sil_state, non_sil_state, function write_fst (line 266) | def write_fst(lexicon, silprobs, sil_phone, sil_disambig, function read_nonterminals (line 356) | def read_nonterminals(filename): function read_left_context_phones (line 372) | def read_left_context_phones(filename): function main (line 387) | def main(): FILE: egs/utils/lang/make_phone_lm.py class CountsForHistory (line 69) | class CountsForHistory(object): method __init__ (line 74) | def __init__(self): method Words (line 80) | def Words(self): method __str__ (line 83) | def __str__(self): method AddCount (line 99) | def AddCount(self, predicted_word, count): class NgramCounts (line 113) | class NgramCounts(object): method __init__ (line 122) | def __init__(self, ngram_order): method AddCount (line 145) | def AddCount(self, history, predicted_word, count): method AddRawCountsFromLine (line 151) | def AddRawCountsFromLine(self, line): method AddRawCountsFromStandardInput (line 165) | def AddRawCountsFromStandardInput(self): method ApplyBackoff (line 184) | def ApplyBackoff(self): method Print (line 218) | def Print(self, info_string): method GetHistToStateMap (line 233) | def GetHistToStateMap(self): method GetProb (line 249) | def GetProb(self, hist, word): method PruneEmptyStates (line 270) | def PruneEmptyStates(self): method EnsureStructurallyNeededNgramsExist (line 300) | def EnsureStructurallyNeededNgramsExist(self): method PrintAsFst (line 339) | def PrintAsFst(self, word_disambig_symbol): method GetProtectedNgrams (line 384) | def GetProtectedNgrams(self): method PruneNgram (line 409) | def PruneNgram(self, hist, word): method PruningLogprobChange (line 442) | def PruningLogprobChange(self, count, discount, backoff_count, backoff... method GetLikeChangeFromPruningNgram (line 547) | def GetLikeChangeFromPruningNgram(self, hist, word): method PruneToIntermediateTarget (line 568) | def PruneToIntermediateTarget(self, num_extra_ngrams): method PruneToFinalTarget (line 643) | def PruneToFinalTarget(self, num_extra_ngrams): method GetNumExtraNgrams (line 694) | def GetNumExtraNgrams(self): method GetNumNgrams (line 702) | def GetNumNgrams(self, hist_len = None): method IntToString (line 721) | def IntToString(self, i): method PrintAsArpa (line 732) | def PrintAsArpa(self): FILE: egs/utils/lang/make_position_dependent_subword_lexicon.py function get_args (line 22) | def get_args(): function is_end (line 39) | def is_end(subword, separator): function write_position_dependent_lexicon (line 44) | def write_position_dependent_lexicon(lexiconp, separator): function main (line 101) | def main(): FILE: egs/utils/lang/make_subword_lexicon_fst.py function get_args (line 12) | def get_args(): function contain_disambig_symbol (line 48) | def contain_disambig_symbol(phones): function print_arc (line 55) | def print_arc(src, dest, phone, word, cost): function is_end (line 58) | def is_end(word, separator): function get_suffix (line 63) | def get_suffix(phone): function write_fst_no_silence (line 71) | def write_fst_no_silence(lexicon, position_dependent, separator): function write_fst_with_silence (line 162) | def write_fst_with_silence(lexicon, sil_phone, sil_prob, sil_disambig, p... function main (line 287) | def main(): FILE: egs/utils/nnet/gen_dct_mat.py function print_on_same_line (line 31) | def print_on_same_line(text): FILE: egs/utils/nnet/gen_hamm_mat.py function print_on_same_line (line 29) | def print_on_same_line(text): FILE: egs/utils/nnet/gen_splice.py function print_on_same_line (line 28) | def print_on_same_line(text): FILE: egs/utils/nnet/make_nnet_proto.py function Glorot (line 105) | def Glorot(dim1, dim2): FILE: egs/utils/nnet3/convert_config_tdnn_to_affine.py function get_parser (line 12) | def get_parser(): function main (line 30) | def main(args): FILE: lm/chainer_backend/extlm.py class MultiLevelLM (line 15) | class MultiLevelLM(chainer.Chain): method __init__ (line 19) | def __init__( method __call__ (line 45) | def __call__(self, state, x): method final (line 95) | def final(self, state): class LookAheadWordLM (line 106) | class LookAheadWordLM(chainer.Chain): method __init__ (line 110) | def __init__( method __call__ (line 127) | def __call__(self, state, x): method final (line 192) | def final(self, state): FILE: lm/chainer_backend/lm.py class DefaultRNNLM (line 51) | class DefaultRNNLM(LMInterface, link.Chain): method add_arguments (line 60) | def add_arguments(parser): class ClassifierWithState (line 78) | class ClassifierWithState(link.Chain): method __init__ (line 86) | def __init__( method __call__ (line 104) | def __call__(self, state, *args, **kwargs): method predict (line 145) | def predict(self, state, x): method final (line 159) | def final(self, state): class RNNLM (line 174) | class RNNLM(chainer.Chain): method __init__ (line 183) | def __init__(self, n_vocab, n_layers, n_units, typ="lstm"): method __call__ (line 204) | def __call__(self, state, x): class BPTTUpdater (line 244) | class BPTTUpdater(training.updaters.StandardUpdater): method __init__ (line 254) | def __init__(self, train_iter, optimizer, schedulers, device, accum_gr... method update_core (line 260) | def update_core(self): class LMEvaluator (line 301) | class LMEvaluator(BaseEvaluator): method __init__ (line 309) | def __init__(self, val_iter, eval_model, device): method evaluate (line 312) | def evaluate(self): function train (line 333) | def train(args): FILE: lm/lm_utils.py function load_dataset (line 21) | def load_dataset(path, label_dict, outdir=None): function read_tokens (line 61) | def read_tokens(filename, label_dict): function count_tokens (line 81) | def count_tokens(data, unk_id=None): function compute_perplexity (line 102) | def compute_perplexity(result): class ParallelSentenceIterator (line 113) | class ParallelSentenceIterator(chainer.dataset.Iterator): method __init__ (line 122) | def __init__( method __next__ (line 165) | def __next__(self): method start_shuffle (line 194) | def start_shuffle(self): method epoch_detail (line 198) | def epoch_detail(self): method previous_epoch_detail (line 203) | def previous_epoch_detail(self): method serialize (line 208) | def serialize(self, serializer): class MakeSymlinkToBestModel (line 227) | class MakeSymlinkToBestModel(extension.Extension): method __init__ (line 235) | def __init__(self, key, prefix="model", suffix="best"): method __call__ (line 243) | def __call__(self, trainer): method serialize (line 257) | def serialize(self, serializer): function make_lexical_tree (line 274) | def make_lexical_tree(word_dict, subword_dict, word_unk): FILE: lm/pytorch_backend/extlm.py class MultiLevelLM (line 18) | class MultiLevelLM(nn.Module): method __init__ (line 22) | def __init__( method forward (line 50) | def forward(self, state, x): method final (line 107) | def final(self, state): class LookAheadWordLM (line 118) | class LookAheadWordLM(nn.Module): method __init__ (line 122) | def __init__( method forward (line 142) | def forward(self, state, x): method final (line 215) | def final(self, state): FILE: lm/pytorch_backend/lm.py function compute_perplexity (line 50) | def compute_perplexity(result): class Reporter (line 63) | class Reporter(Chain): method report (line 66) | def report(self, loss): function concat_examples (line 71) | def concat_examples(batch, device=None, padding=None): class BPTTUpdater (line 89) | class BPTTUpdater(training.StandardUpdater): method __init__ (line 92) | def __init__( method update_core (line 126) | def update_core(self): class LMEvaluator (line 169) | class LMEvaluator(BaseEvaluator): method __init__ (line 172) | def __init__(self, val_iter, eval_model, reporter, device): method evaluate (line 185) | def evaluate(self): function train (line 213) | def train(args): FILE: mt/mt_utils.py function parse_hypothesis (line 13) | def parse_hypothesis(hyp, char_list): function add_results_to_json (line 35) | def add_results_to_json(js, nbest_hyps, char_list): FILE: mt/pytorch_backend/mt.py class CustomConverter (line 56) | class CustomConverter(object): method __init__ (line 59) | def __init__(self): method __call__ (line 68) | def __call__(self, batch, device=torch.device("cpu")): function train (line 96) | def train(args): function trans (line 514) | def trans(args): FILE: nets/asr_interface.py class ASRInterface (line 9) | class ASRInterface: method add_arguments (line 13) | def add_arguments(parser): method build (line 18) | def build(cls, idim: int, odim: int, **kwargs): method forward (line 38) | def forward(self, xs, ilens, ys): method recognize (line 55) | def recognize(self, x, recog_args, char_list=None, rnnlm=None): method recognize_batch (line 67) | def recognize_batch(self, x, recog_args, char_list=None, rnnlm=None): method calculate_all_attentions (line 79) | def calculate_all_attentions(self, xs, ilens, ys): method calculate_all_ctc_probs (line 90) | def calculate_all_ctc_probs(self, xs, ilens, ys): method attention_plot_class (line 102) | def attention_plot_class(self): method ctc_plot_class (line 109) | def ctc_plot_class(self): method get_total_subsampling_factor (line 115) | def get_total_subsampling_factor(self): method encode (line 121) | def encode(self, feat): method scorers (line 133) | def scorers(self): function dynamic_import_asr (line 157) | def dynamic_import_asr(module, backend): FILE: nets/batch_beam_search.py class BatchHypothesis (line 17) | class BatchHypothesis(NamedTuple): method __len__ (line 26) | def __len__(self) -> int: class BatchBeamSearch (line 31) | class BatchBeamSearch(BeamSearch): method batchfy (line 34) | def batchfy(self, hyps: List[Hypothesis]) -> BatchHypothesis: method _batch_select (line 48) | def _batch_select(self, hyps: BatchHypothesis, ids: List[int]) -> Batc... method _select (line 60) | def _select(self, hyps: BatchHypothesis, i: int) -> Hypothesis: method unbatchfy (line 70) | def unbatchfy(self, batch_hyps: BatchHypothesis) -> List[Hypothesis]: method batch_beam (line 85) | def batch_beam( method init_hyp (line 111) | def init_hyp(self, x: torch.Tensor) -> BatchHypothesis: method score_full (line 137) | def score_full( method score_partial (line 160) | def score_partial( method merge_states (line 186) | def merge_states(self, states: Any, part_states: Any, part_idx: int) -... method search (line 207) | def search(self, running_hyps: BatchHypothesis, x: torch.Tensor) -> Ba... method post_process (line 286) | def post_process( FILE: nets/batch_beam_search_online_sim.py class BatchBeamSearchOnlineSim (line 16) | class BatchBeamSearchOnlineSim(BatchBeamSearch): method set_streaming_config (line 28) | def set_streaming_config(self, asr_config: str): method set_block_size (line 72) | def set_block_size(self, block_size: int): method set_hop_size (line 80) | def set_hop_size(self, hop_size: int): method set_look_ahead (line 88) | def set_look_ahead(self, look_ahead: int): method forward (line 96) | def forward( method extend (line 255) | def extend(self, x: torch.Tensor, hyps: Hypothesis) -> List[Hypothesis]: FILE: nets/beam_search.py class Hypothesis (line 19) | class Hypothesis(NamedTuple): method asdict (line 27) | def asdict(self) -> dict: method __str__ (line 35) | def __str__(self): class BeamSearch (line 46) | class BeamSearch(object): method __init__ (line 49) | def __init__( method init_hyp (line 131) | def init_hyp(self, x: torch.Tensor) -> List[Hypothesis]: method append_token (line 156) | def append_token(xs: torch.Tensor, x: int) -> torch.Tensor: method score_full (line 170) | def score_full( method score_partial (line 193) | def score_partial( method beam (line 217) | def beam( method merge_scores (line 247) | def merge_scores( method merge_states (line 278) | def merge_states(self, states: Any, part_states: Any, part_idx: int) -... method search (line 299) | def search( method __call__ (line 357) | def __call__( method post_process (line 445) | def post_process( function beam_search (line 496) | def beam_search( FILE: nets/beam_search_transducer.py class BeamSearchTransducer (line 25) | class BeamSearchTransducer: method __init__ (line 28) | def __init__( method __call__ (line 162) | def __call__(self, h: torch.Tensor) -> Union[List[Hypothesis], List[NS... method sort_nbest (line 186) | def sort_nbest( method vocab_regularization (line 205) | def vocab_regularization(self, hyps): method greedy_search (line 218) | def greedy_search(self, h: torch.Tensor) -> List[Hypothesis]: method ctc_greedy_search (line 247) | def ctc_greedy_search(self, h: torch.Tensor) -> List[Hypothesis]: method ctc_beam_search (line 258) | def ctc_beam_search(self, h: torch.Tensor) -> List[Hypothesis]: method default_beam_search (line 359) | def default_beam_search(self, h: torch.Tensor) -> List[Hypothesis]: method time_sync_decoding (line 431) | def time_sync_decoding(self, h: torch.Tensor) -> List[Hypothesis]: method align_length_sync_decoding (line 531) | def align_length_sync_decoding(self, h: torch.Tensor) -> List[Hypothes... method nsc_beam_search (line 740) | def nsc_beam_search(self, h: torch.Tensor) -> List[NSCHypothesis]: function log_add (line 931) | def log_add(args: List[int]) -> float: function is_all_chinese (line 941) | def is_all_chinese(strs): FILE: nets/chainer_backend/asr_interface.py class ChainerASRInterface (line 7) | class ChainerASRInterface(ASRInterface, chainer.Chain): method custom_converter (line 11) | def custom_converter(*args, **kw): method custom_updater (line 16) | def custom_updater(*args, **kw): method custom_parallel_updater (line 21) | def custom_parallel_updater(*args, **kw): method get_total_subsampling_factor (line 25) | def get_total_subsampling_factor(self): FILE: nets/chainer_backend/ctc.py class CTC (line 10) | class CTC(chainer.Chain): method __init__ (line 20) | def __init__(self, odim, eprojs, dropout_rate): method __call__ (line 28) | def __call__(self, hs, ys): method log_softmax (line 72) | def log_softmax(self, hs): class WarpCTC (line 87) | class WarpCTC(chainer.Chain): method __init__ (line 97) | def __init__(self, odim, eprojs, dropout_rate): method __call__ (line 105) | def __call__(self, hs, ys): method log_softmax (line 140) | def log_softmax(self, hs): method argmax (line 154) | def argmax(self, hs_pad): function ctc_for (line 164) | def ctc_for(args, odim): FILE: nets/chainer_backend/deterministic_embed_id.py class EmbedIDFunction (line 22) | class EmbedIDFunction(function_node.FunctionNode): method __init__ (line 23) | def __init__(self, ignore_label=None): method check_type_forward (line 27) | def check_type_forward(self, in_types): method forward (line 36) | def forward(self, inputs): method backward (line 63) | def backward(self, indexes, grad_outputs): class EmbedIDGrad (line 71) | class EmbedIDGrad(function_node.FunctionNode): method __init__ (line 72) | def __init__(self, w_shape, ignore_label=None): method forward (line 77) | def forward(self, inputs): method backward (line 125) | def backward(self, indexes, grads): function embed_id (line 147) | def embed_id(x, W, ignore_label=None): class EmbedID (line 194) | class EmbedID(link.Link): method __init__ (line 234) | def __init__(self, in_size, out_size, initialW=None, ignore_label=None): method __call__ (line 243) | def __call__(self, x): FILE: nets/chainer_backend/e2e_asr.py class E2E (line 25) | class E2E(ChainerASRInterface): method add_arguments (line 39) | def add_arguments(parser): method get_total_subsampling_factor (line 43) | def get_total_subsampling_factor(self): method __init__ (line 47) | def __init__(self, idim, odim, args, flag_return=True): method forward (line 93) | def forward(self, xs, ilens, ys): method recognize (line 147) | def recognize(self, x, recog_args, char_list, rnnlm=None): method calculate_all_attentions (line 182) | def calculate_all_attentions(self, xs, ilens, ys): method custom_converter (line 200) | def custom_converter(subsampling_factor=0): method custom_updater (line 207) | def custom_updater(iters, optimizer, converter, device=-1, accum_grad=1): method custom_parallel_updater (line 216) | def custom_parallel_updater(iters, optimizer, converter, devices, accu... FILE: nets/chainer_backend/e2e_asr_transformer.py class E2E (line 39) | class E2E(ChainerASRInterface): method add_arguments (line 53) | def add_arguments(parser): method get_total_subsampling_factor (line 142) | def get_total_subsampling_factor(self): method __init__ (line 146) | def __init__(self, idim, odim, args, ignore_id=-1, flag_return=True): method reset_parameters (line 217) | def reset_parameters(self, args): method forward (line 251) | def forward(self, xs, ilens, ys_pad, calculate_attentions=False): method calculate_attentions (line 344) | def calculate_attentions(self, xs, x_mask, ys_pad): method recognize (line 348) | def recognize(self, x_block, recog_args, char_list=None, rnnlm=None): method recognize_beam (line 380) | def recognize_beam(self, h, lpz, recog_args, char_list=None, rnnlm=None): method calculate_all_attentions (line 573) | def calculate_all_attentions(self, xs, ilens, ys): method attention_plot_class (line 598) | def attention_plot_class(self): method custom_converter (line 610) | def custom_converter(subsampling_factor=0): method custom_updater (line 615) | def custom_updater(iters, optimizer, converter, device=-1, accum_grad=1): method custom_parallel_updater (line 622) | def custom_parallel_updater(iters, optimizer, converter, devices, accu... FILE: nets/chainer_backend/nets_utils.py function _subsamplex (line 4) | def _subsamplex(x, n): FILE: nets/chainer_backend/rnn/attentions.py class AttDot (line 9) | class AttDot(chainer.Chain): method __init__ (line 19) | def __init__(self, eprojs, dunits, att_dim): method reset (line 32) | def reset(self): method __call__ (line 38) | def __call__(self, enc_hs, dec_z, att_prev, scaling=2.0): class AttLoc (line 87) | class AttLoc(chainer.Chain): method __init__ (line 99) | def __init__(self, eprojs, dunits, att_dim, aconv_chans, aconv_filts): method reset (line 118) | def reset(self): method __call__ (line 124) | def __call__(self, enc_hs, dec_z, att_prev, scaling=2.0): class NoAtt (line 200) | class NoAtt(chainer.Chain): method __init__ (line 208) | def __init__(self): method reset (line 215) | def reset(self): method __call__ (line 222) | def __call__(self, enc_hs, dec_z, att_prev): function att_for (line 258) | def att_for(args): FILE: nets/chainer_backend/rnn/decoders.py class Decoder (line 21) | class Decoder(chainer.Chain): method __init__ (line 42) | def __init__( method rnn_forward (line 90) | def rnn_forward(self, ey, z_list, c_list, z_prev, c_prev): method __call__ (line 118) | def __call__(self, hs, ys): method recognize_beam (line 221) | def recognize_beam(self, h, lpz, recog_args, char_list, rnnlm=None): method calculate_all_attentions (line 444) | def calculate_all_attentions(self, hs, ys): function decoder_for (line 498) | def decoder_for(args, odim, sos, eos, att, labeldist): FILE: nets/chainer_backend/rnn/encoders.py class RNNP (line 16) | class RNNP(chainer.Chain): method __init__ (line 30) | def __init__(self, idim, elayers, cdim, hdim, subsample, dropout, typ=... method __call__ (line 58) | def __call__(self, xs, ilens): class RNN (line 94) | class RNN(chainer.Chain): method __init__ (line 107) | def __init__(self, idim, elayers, cdim, hdim, dropout, typ="lstm"): method __call__ (line 121) | def __call__(self, xs, ilens): class VGG2L (line 155) | class VGG2L(chainer.Chain): method __init__ (line 163) | def __init__(self, in_channel=1): method __call__ (line 174) | def __call__(self, xs, ilens): class Encoder (line 227) | class Encoder(chainer.Chain): method __init__ (line 241) | def __init__( method __call__ (line 293) | def __call__(self, xs, ilens): function encoder_for (line 310) | def encoder_for(args, idim, subsample): FILE: nets/chainer_backend/rnn/training.py function sum_sqnorm (line 23) | def sum_sqnorm(arr): class CustomUpdater (line 43) | class CustomUpdater(training.StandardUpdater): method __init__ (line 68) | def __init__(self, train_iter, optimizer, converter, device, accum_gra... method update_core (line 79) | def update_core(self): method update (line 112) | def update(self): class CustomParallelUpdater (line 118) | class CustomParallelUpdater(training.updaters.MultiprocessParallelUpdater): method __init__ (line 145) | def __init__(self, train_iters, optimizer, converter, devices, accum_g... method update_core (line 156) | def update_core(self): method update (line 212) | def update(self): class CustomConverter (line 218) | class CustomConverter(object): method __init__ (line 226) | def __init__(self, subsampling_factor=1): method __call__ (line 229) | def __call__(self, batch, device): FILE: nets/chainer_backend/transformer/attention.py class MultiHeadAttention (line 14) | class MultiHeadAttention(chainer.Chain): method __init__ (line 30) | def __init__(self, n_units, h=8, dropout=0.1, initialW=None, initial_b... method forward (line 65) | def forward(self, e_var, s_var=None, mask=None, batch=1): FILE: nets/chainer_backend/transformer/ctc.py class CTC (line 12) | class CTC(chainer.Chain): method __init__ (line 22) | def __init__(self, odim, eprojs, dropout_rate): method __call__ (line 31) | def __call__(self, hs, ys): method log_softmax (line 75) | def log_softmax(self, hs): class WarpCTC (line 90) | class WarpCTC(chainer.Chain): method __init__ (line 100) | def __init__(self, odim, eprojs, dropout_rate): method forward (line 117) | def forward(self, hs, ys): method log_softmax (line 148) | def log_softmax(self, hs): method argmax (line 162) | def argmax(self, hs_pad): FILE: nets/chainer_backend/transformer/decoder.py class Decoder (line 17) | class Decoder(chainer.Chain): method __init__ (line 32) | def __init__(self, odim, args, initialW=None, initial_bias=None): method make_attention_mask (line 70) | def make_attention_mask(self, source_block, target_block): method forward (line 84) | def forward(self, ys_pad, source, x_mask): method recognize (line 112) | def recognize(self, e, yy_mask, source): FILE: nets/chainer_backend/transformer/decoder_layer.py class DecoderLayer (line 15) | class DecoderLayer(chainer.Chain): method __init__ (line 26) | def __init__( method forward (line 58) | def forward(self, e, s, xy_mask, yy_mask, batch): FILE: nets/chainer_backend/transformer/embedding.py class PositionalEncoding (line 10) | class PositionalEncoding(chainer.Chain): method __init__ (line 19) | def __init__(self, n_units, dropout=0.1, length=5000): method forward (line 33) | def forward(self, e): FILE: nets/chainer_backend/transformer/encoder.py class Encoder (line 19) | class Encoder(chainer.Chain): method __init__ (line 34) | def __init__( method forward (line 104) | def forward(self, e, ilens): FILE: nets/chainer_backend/transformer/encoder_layer.py class EncoderLayer (line 15) | class EncoderLayer(chainer.Chain): method __init__ (line 26) | def __init__( method forward (line 51) | def forward(self, e, xx_mask, batch): FILE: nets/chainer_backend/transformer/label_smoothing_loss.py class LabelSmoothingLoss (line 11) | class LabelSmoothingLoss(chainer.Chain): method __init__ (line 21) | def __init__(self, smoothing, n_target_vocab, normalize_length=False, ... method forward (line 35) | def forward(self, ys_block, ys_pad): FILE: nets/chainer_backend/transformer/layer_norm.py class LayerNorm (line 7) | class LayerNorm(L.LayerNormalization): method __init__ (line 10) | def __init__(self, dims, eps=1e-12): method __call__ (line 14) | def __call__(self, e): FILE: nets/chainer_backend/transformer/mask.py function make_history_mask (line 4) | def make_history_mask(xp, block): FILE: nets/chainer_backend/transformer/positionwise_feed_forward.py class PositionwiseFeedForward (line 12) | class PositionwiseFeedForward(chainer.Chain): method __init__ (line 22) | def __init__( method __call__ (line 55) | def __call__(self, e): FILE: nets/chainer_backend/transformer/subsampling.py class Conv2dSubsampling (line 15) | class Conv2dSubsampling(chainer.Chain): method __init__ (line 24) | def __init__( method forward (line 63) | def forward(self, xs, ilens): class LinearSampling (line 82) | class LinearSampling(chainer.Chain): method __init__ (line 91) | def __init__(self, idim, dims, dropout=0.1, initialW=None, initial_bia... method forward (line 105) | def forward(self, xs, ilens): FILE: nets/chainer_backend/transformer/training.py function sum_sqnorm (line 20) | def sum_sqnorm(arr): class CustomUpdater (line 40) | class CustomUpdater(training.StandardUpdater): method __init__ (line 65) | def __init__(self, train_iter, optimizer, converter, device, accum_gra... method update_core (line 77) | def update_core(self): method update (line 108) | def update(self): class CustomParallelUpdater (line 115) | class CustomParallelUpdater(training.updaters.MultiprocessParallelUpdater): method __init__ (line 141) | def __init__(self, train_iters, optimizer, converter, devices, accum_g... method update_core (line 154) | def update_core(self): method update (line 208) | def update(self): class VaswaniRule (line 215) | class VaswaniRule(extension.Extension): method __init__ (line 233) | def __init__( method initialize (line 253) | def initialize(self, trainer): method __call__ (line 264) | def __call__(self, trainer): method serialize (line 273) | def serialize(self, serializer): method _get_optimizer (line 278) | def _get_optimizer(self, trainer): method _update_value (line 282) | def _update_value(self, optimizer, value): class CustomConverter (line 288) | class CustomConverter(object): method __init__ (line 296) | def __init__(self): method __call__ (line 300) | def __call__(self, batch, device): FILE: nets/ctc_prefix_score.py class CTCPrefixScoreTH (line 12) | class CTCPrefixScoreTH(object): method __init__ (line 23) | def __init__(self, x, xlens, blank, eos, margin=0): method __call__ (line 69) | def __call__(self, y, state, scoring_ids=None, att_w=None): method index_select_state (line 190) | def index_select_state(self, state, best_ids): method extend_prob (line 223) | def extend_prob(self, x): method extend_state (line 245) | def extend_state(self, state): class CTCPrefixScore (line 273) | class CTCPrefixScore(object): method __init__ (line 284) | def __init__(self, x, blank, eos, xp): method initial_state (line 292) | def initial_state(self): method __call__ (line 306) | def __call__(self, y, cs, r_prev): FILE: nets/e2e_asr_common.py function end_detect (line 19) | def end_detect(ended_hyps, i, M=3, D_end=np.log(1 * np.exp(-10))): function label_smoothing_dist (line 53) | def label_smoothing_dist(odim, lsm_type, transcript=None, blank=0): function get_vgg2l_odim (line 87) | def get_vgg2l_odim(idim, in_channel=3, out_channel=128): class ErrorCalculator (line 101) | class ErrorCalculator(object): method __init__ (line 112) | def __init__( method __call__ (line 130) | def __call__(self, ys_hat, ys_pad, is_ctc=False): method calculate_cer_ctc (line 155) | def calculate_cer_ctc(self, ys_hat, ys_pad): method convert_to_char (line 187) | def convert_to_char(self, ys_hat, ys_pad): method calculate_cer (line 212) | def calculate_cer(self, seqs_hat, seqs_true): method calculate_wer (line 229) | def calculate_wer(self, seqs_hat, seqs_true): FILE: nets/e2e_mt_common.py class ErrorCalculator (line 13) | class ErrorCalculator(object): method __init__ (line 24) | def __init__(self, char_list, sym_space, sym_pad, report_bleu=False): method __call__ (line 36) | def __call__(self, ys_hat, ys_pad): method calculate_corpus_bleu (line 51) | def calculate_corpus_bleu(self, ys_hat, ys_pad): FILE: nets/lm_interface.py class LMInterface (line 10) | class LMInterface(ScorerInterface): method add_arguments (line 14) | def add_arguments(parser): method build (line 19) | def build(cls, n_vocab: int, **kwargs): method forward (line 40) | def forward(self, x, t): function dynamic_import_lm (line 71) | def dynamic_import_lm(module, backend): FILE: nets/mt_interface.py class MTInterface (line 8) | class MTInterface: method add_arguments (line 12) | def add_arguments(parser): method build (line 17) | def build(cls, idim: int, odim: int, **kwargs): method forward (line 37) | def forward(self, xs, ilens, ys): method translate (line 54) | def translate(self, x, trans_args, char_list=None, rnnlm=None): method translate_batch (line 66) | def translate_batch(self, x, trans_args, char_list=None, rnnlm=None): method calculate_all_attentions (line 78) | def calculate_all_attentions(self, xs, ilens, ys): method attention_plot_class (line 90) | def attention_plot_class(self): FILE: nets/pytorch_backend/conformer/argument.py function add_arguments_conformer_common (line 11) | def add_arguments_conformer_common(group): function verify_rel_pos_type (line 66) | def verify_rel_pos_type(args): FILE: nets/pytorch_backend/conformer/convolution.py class ConvolutionModule (line 13) | class ConvolutionModule(nn.Module): method __init__ (line 22) | def __init__(self, channels, kernel_size, activation=nn.ReLU(), bias=T... method forward (line 59) | def forward(self, x): FILE: nets/pytorch_backend/conformer/encoder.py class Encoder (line 35) | class Encoder(torch.nn.Module): method __init__ (line 66) | def __init__( method forward (line 222) | def forward(self, xs, masks): FILE: nets/pytorch_backend/conformer/encoder_layer.py class EncoderLayer (line 17) | class EncoderLayer(nn.Module): method __init__ (line 42) | def __init__( method forward (line 76) | def forward(self, x_input, mask, cache=None): FILE: nets/pytorch_backend/conformer/swish.py class Swish (line 13) | class Swish(torch.nn.Module): method forward (line 16) | def forward(self, x): FILE: nets/pytorch_backend/ctc.py class CTC (line 12) | class CTC(torch.nn.Module): method __init__ (line 22) | def __init__(self, odim, eprojs, dropout_rate, ctc_type="warpctc", red... method loss_fn (line 69) | def loss_fn(self, th_pred, th_target, th_ilen, th_olen): method forward (line 89) | def forward(self, hs_pad, hlens, ys_pad, texts): method softmax (line 158) | def softmax(self, hs_pad): method log_softmax (line 168) | def log_softmax(self, hs_pad): method argmax (line 177) | def argmax(self, hs_pad): method forced_align (line 186) | def forced_align(self, h, y, blank_id=0): function ctc_for (line 252) | def ctc_for(args, odim, reduce=True): FILE: nets/pytorch_backend/e2e_asr.py class Reporter (line 43) | class Reporter(chainer.Chain): method report (line 46) | def report(self, loss_ctc, loss_att, loss_third, loss_mbr, acc, cer_ct... class E2E (line 60) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 70) | def add_arguments(parser): method encoder_add_arguments (line 78) | def encoder_add_arguments(parser): method attention_add_arguments (line 85) | def attention_add_arguments(parser): method decoder_add_arguments (line 92) | def decoder_add_arguments(parser): method get_total_subsampling_factor (line 98) | def get_total_subsampling_factor(self): method __init__ (line 105) | def __init__(self, idim, odim, args): method init_like_chainer (line 186) | def init_like_chainer(self): method forward (line 204) | def forward(self, xs_pad, ilens, ys_pad): method scorers (line 339) | def scorers(self): method encode (line 343) | def encode(self, x): method recognize (line 371) | def recognize(self, x, recog_args, char_list, rnnlm=None): method recognize_batch (line 393) | def recognize_batch(self, xs, recog_args, char_list, rnnlm=None): method enhance (line 446) | def enhance(self, xs): method calculate_all_attentions (line 468) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad): method calculate_all_ctc_probs (line 496) | def calculate_all_ctc_probs(self, xs_pad, ilens, ys_pad): method subsample_frames (line 526) | def subsample_frames(self, x): FILE: nets/pytorch_backend/e2e_asr_conformer.py class E2E (line 21) | class E2E(E2ETransformer): method add_arguments (line 31) | def add_arguments(parser): method add_conformer_arguments (line 38) | def add_conformer_arguments(parser): method __init__ (line 44) | def __init__(self, idim, odim, args, ignore_id=-1): FILE: nets/pytorch_backend/e2e_asr_maskctc.py class E2E (line 32) | class E2E(E2ETransformer): method add_arguments (line 42) | def add_arguments(parser): method add_maskctc_arguments (line 50) | def add_maskctc_arguments(parser): method __init__ (line 63) | def __init__(self, idim, odim, args, ignore_id=-1): method forward (line 102) | def forward(self, xs_pad, ilens, ys_pad): method recognize (line 174) | def recognize(self, x, recog_args, char_list=None, rnnlm=None): FILE: nets/pytorch_backend/e2e_asr_mix.py class PIT (line 48) | class PIT(object): method __init__ (line 54) | def __init__(self, num_spkrs): method min_pit_sample (line 71) | def min_pit_sample(self, loss): method pit_process (line 93) | def pit_process(self, losses): method permutationDFS (line 110) | def permutationDFS(self, source, start): class E2E (line 131) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 140) | def add_arguments(parser): method encoder_mix_add_arguments (line 149) | def encoder_mix_add_arguments(parser): method get_total_subsampling_factor (line 168) | def get_total_subsampling_factor(self): method __init__ (line 172) | def __init__(self, idim, odim, args): method init_like_chainer (line 255) | def init_like_chainer(self): method forward (line 274) | def forward(self, xs_pad, ilens, ys_pad): method recognize (line 491) | def recognize(self, x, recog_args, char_list, rnnlm=None): method recognize_batch (line 547) | def recognize_batch(self, xs, recog_args, char_list, rnnlm=None): method enhance (line 610) | def enhance(self, xs): method calculate_all_attentions (line 638) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad): class EncoderMix (line 697) | class EncoderMix(torch.nn.Module): method __init__ (line 714) | def __init__( method forward (line 782) | def forward(self, xs_pad, ilens): function encoder_for (line 811) | def encoder_for(args, idim, subsample): FILE: nets/pytorch_backend/e2e_asr_mix_transformer.py class E2E (line 43) | class E2E(E2EASR, ASRInterface, torch.nn.Module): method add_arguments (line 52) | def add_arguments(parser): method __init__ (line 58) | def __init__(self, idim, odim, args, ignore_id=-1): method forward (line 92) | def forward(self, xs_pad, ilens, ys_pad): method decoder_and_attention (line 212) | def decoder_and_attention(self, hs_pad, hs_mask, ys_pad, batch_size): method encode (line 226) | def encode(self, x): method recog (line 238) | def recog(self, enc_output, recog_args, char_list=None, rnnlm=None, us... method recognize (line 443) | def recognize(self, x, recog_args, char_list=None, rnnlm=None, use_jit... FILE: nets/pytorch_backend/e2e_asr_mulenc.py class Reporter (line 36) | class Reporter(chainer.Chain): method report (line 39) | def report(self, loss_ctc_list, loss_att, acc, cer_ctc_list, cer, wer,... class E2E (line 58) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 68) | def add_arguments(parser): method encoder_add_arguments (line 77) | def encoder_add_arguments(parser): method attention_add_arguments (line 132) | def attention_add_arguments(parser): method decoder_add_arguments (line 250) | def decoder_add_arguments(parser): method ctc_add_arguments (line 290) | def ctc_add_arguments(parser): method get_total_subsampling_factor (line 314) | def get_total_subsampling_factor(self): method __init__ (line 325) | def __init__(self, idims, odim, args): method init_like_chainer (line 438) | def init_like_chainer(self): method forward (line 484) | def forward(self, xs_pad_list, ilens_list, ys_pad): method scorers (line 673) | def scorers(self): method encode (line 682) | def encode(self, x_list): method recognize (line 717) | def recognize(self, x_list, recog_args, char_list, rnnlm=None): method recognize_batch (line 748) | def recognize_batch(self, xs_list, recog_args, char_list, rnnlm=None): method calculate_all_attentions (line 821) | def calculate_all_attentions(self, xs_pad_list, ilens_list, ys_pad): method calculate_all_ctc_probs (line 859) | def calculate_all_ctc_probs(self, xs_pad_list, ilens_list, ys_pad): FILE: nets/pytorch_backend/e2e_asr_transducer.py class Reporter (line 62) | class Reporter(chainer.Chain): method report (line 65) | def report( class E2E (line 95) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 108) | def add_arguments(parser): method att_scorer_arguments (line 126) | def att_scorer_arguments(parser): method encoder_add_general_arguments (line 134) | def encoder_add_general_arguments(parser): method encoder_add_rnn_arguments (line 142) | def encoder_add_rnn_arguments(parser): method encoder_add_custom_arguments (line 150) | def encoder_add_custom_arguments(parser): method decoder_add_general_arguments (line 158) | def decoder_add_general_arguments(parser): method decoder_add_rnn_arguments (line 166) | def decoder_add_rnn_arguments(parser): method decoder_add_custom_arguments (line 174) | def decoder_add_custom_arguments(parser): method training_add_custom_arguments (line 182) | def training_add_custom_arguments(parser): method transducer_add_arguments (line 190) | def transducer_add_arguments(parser): method auxiliary_task_add_arguments (line 198) | def auxiliary_task_add_arguments(parser): method attention_plot_class (line 206) | def attention_plot_class(self): method get_total_subsampling_factor (line 210) | def get_total_subsampling_factor(self): method __init__ (line 219) | def __init__(self, idim, odim, args, ignore_id=-1, blank_id=0, trainin... method default_parameters (line 472) | def default_parameters(self, args): method forward (line 481) | def forward(self, xs_pad, ilens, ys_pad, texts, xs_pad_orig): method mbr_forward (line 664) | def mbr_forward(self, xs_pad_orig, ilens, ys_pad, hs_pad): method compute_edit_distance (line 776) | def compute_edit_distance(self, hyps, refs): method encode_custom (line 796) | def encode_custom(self, x): method encode_rnn (line 811) | def encode_rnn(self, x): method recognize (line 833) | def recognize(self, x, beam_search, decode_feature="combine"): method calculate_all_attentions (line 856) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad, texts, xs_pa... FILE: nets/pytorch_backend/e2e_asr_transducer_cs.py class Reporter (line 47) | class Reporter(chainer.Chain): method report (line 50) | def report( class E2E (line 81) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 94) | def add_arguments(parser): method encoder_add_general_arguments (line 110) | def encoder_add_general_arguments(parser): method encoder_add_custom_arguments (line 118) | def encoder_add_custom_arguments(parser): method decoder_add_general_arguments (line 126) | def decoder_add_general_arguments(parser): method decoder_add_rnn_arguments (line 134) | def decoder_add_rnn_arguments(parser): method training_add_custom_arguments (line 142) | def training_add_custom_arguments(parser): method transducer_add_arguments (line 150) | def transducer_add_arguments(parser): method transducer_add_code_switch_arguments (line 158) | def transducer_add_code_switch_arguments(parser): method auxiliary_task_add_arguments (line 166) | def auxiliary_task_add_arguments(parser): method get_total_subsampling_factor (line 173) | def get_total_subsampling_factor(self): method __init__ (line 184) | def __init__(self, idim, odim, args, ignore_id=-1, blank_id=0, trainin... method default_parameters (line 324) | def default_parameters(self, args): method forward (line 334) | def forward(self, xs_pad, ilens, ys_pad, texts, xs_pad_orig): method combine_fn (line 439) | def combine_fn(self, chn_hs_pad, eng_hs_pad, chn_hs_mask, eng_hs_mask): method monolingual_mask (line 442) | def monolingual_mask(self, ys_pad): method encoder_forward (line 455) | def encoder_forward(self, x): method recognize (line 478) | def recognize(self, x, beam_search=None, decode_feature="combine"): method calculate_all_attentions (line 493) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad, texts, xs_pa... FILE: nets/pytorch_backend/e2e_asr_transformer.py class E2E (line 60) | class E2E(ASRInterface, torch.nn.Module): method add_arguments (line 70) | def add_arguments(parser): method attention_plot_class (line 79) | def attention_plot_class(self): method get_total_subsampling_factor (line 83) | def get_total_subsampling_factor(self): method __init__ (line 87) | def __init__(self, idim, odim, args, ignore_id=-1): method reset_parameters (line 218) | def reset_parameters(self, args): method forward (line 223) | def forward(self, xs_pad, ilens, ys_pad, texts, xs_pad_orig): method mbr_forward (line 328) | def mbr_forward(self, xs_pad_orig, ilens, ys_pad, hs_pad, hs_mask): method compute_edit_distance (line 392) | def compute_edit_distance(self, hyps, refs): method scorers (line 412) | def scorers(self): method encode (line 416) | def encode(self, x): method recognize (line 428) | def recognize(self, x, recog_args, char_list=None, rnnlm=None, use_jit... method calculate_all_attentions (line 755) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad, texts, xs_pa... method calculate_all_ctc_probs (line 781) | def calculate_all_ctc_probs(self, xs_pad, ilens, ys_pad, texts, xs_pad... FILE: nets/pytorch_backend/e2e_mt.py class Reporter (line 34) | class Reporter(chainer.Chain): method report (line 37) | def report(self, loss, acc, ppl, bleu): class E2E (line 45) | class E2E(MTInterface, torch.nn.Module): method add_arguments (line 55) | def add_arguments(parser): method encoder_add_arguments (line 63) | def encoder_add_arguments(parser): method attention_add_arguments (line 70) | def attention_add_arguments(parser): method decoder_add_arguments (line 77) | def decoder_add_arguments(parser): method __init__ (line 83) | def __init__(self, idim, odim, args): method init_like_fairseq (line 190) | def init_like_fairseq(self): method forward (line 203) | def forward(self, xs_pad, ilens, ys_pad): method target_language_biasing (line 260) | def target_language_biasing(self, xs_pad, ilens, ys_pad): method translate (line 284) | def translate(self, x, trans_args, char_list, rnnlm=None): method translate_batch (line 319) | def translate_batch(self, xs, trans_args, char_list, rnnlm=None): method calculate_all_attentions (line 353) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad): FILE: nets/pytorch_backend/e2e_mt_transformer.py class E2E (line 38) | class E2E(MTInterface, torch.nn.Module): method add_arguments (line 48) | def add_arguments(parser): method attention_plot_class (line 55) | def attention_plot_class(self): method __init__ (line 59) | def __init__(self, idim, odim, args, ignore_id=-1): method reset_parameters (line 140) | def reset_parameters(self, args): method forward (line 152) | def forward(self, xs_pad, ilens, ys_pad): method scorers (line 203) | def scorers(self): method encode (line 207) | def encode(self, xs): method target_forcing (line 214) | def target_forcing(self, xs_pad, ys_pad=None, tgt_lang=None): method translate (line 242) | def translate(self, x, trans_args, char_list=None): method calculate_all_attentions (line 393) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad): FILE: nets/pytorch_backend/e2e_st.py class Reporter (line 46) | class Reporter(chainer.Chain): method report (line 49) | def report( class E2E (line 78) | class E2E(STInterface, torch.nn.Module): method add_arguments (line 88) | def add_arguments(parser): method encoder_add_arguments (line 96) | def encoder_add_arguments(parser): method attention_add_arguments (line 103) | def attention_add_arguments(parser): method decoder_add_arguments (line 110) | def decoder_add_arguments(parser): method get_total_subsampling_factor (line 116) | def get_total_subsampling_factor(self): method __init__ (line 120) | def __init__(self, idim, odim, args): method init_like_chainer (line 263) | def init_like_chainer(self): method forward (line 281) | def forward(self, xs_pad, ilens, ys_pad, ys_pad_src): method forward_asr (line 387) | def forward_asr(self, hs_pad, hlens, ys_pad): method forward_mt (line 502) | def forward_mt(self, xs_pad, ys_pad): method scorers (line 527) | def scorers(self): method encode (line 531) | def encode(self, x): method translate (line 552) | def translate(self, x, trans_args, char_list, rnnlm=None): method translate_batch (line 571) | def translate_batch(self, xs, trans_args, char_list, rnnlm=None): method calculate_all_attentions (line 603) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad, ys_pad_src): method calculate_all_ctc_probs (line 633) | def calculate_all_ctc_probs(self, xs_pad, ilens, ys_pad, ys_pad_src): method subsample_frames (line 658) | def subsample_frames(self, x): FILE: nets/pytorch_backend/e2e_st_conformer.py class E2E (line 20) | class E2E(E2ETransformer): method add_arguments (line 30) | def add_arguments(parser): method add_conformer_arguments (line 37) | def add_conformer_arguments(parser): method __init__ (line 43) | def __init__(self, idim, odim, args, ignore_id=-1): FILE: nets/pytorch_backend/e2e_st_transformer.py class E2E (line 41) | class E2E(STInterface, torch.nn.Module): method add_arguments (line 51) | def add_arguments(parser): method attention_plot_class (line 58) | def attention_plot_class(self): method get_total_subsampling_factor (line 62) | def get_total_subsampling_factor(self): method __init__ (line 66) | def __init__(self, idim, odim, args, ignore_id=-1): method reset_parameters (line 183) | def reset_parameters(self, args): method forward (line 192) | def forward(self, xs_pad, ilens, ys_pad, ys_pad_src): method forward_asr (line 283) | def forward_asr(self, hs_pad, hs_mask, ys_pad): method forward_mt (line 343) | def forward_mt(self, xs_pad, ys_in_pad, ys_out_pad, ys_mask): method scorers (line 375) | def scorers(self): method encode (line 379) | def encode(self, x): method translate (line 391) | def translate( method calculate_all_attentions (line 540) | def calculate_all_attentions(self, xs_pad, ilens, ys_pad, ys_pad_src): method calculate_all_ctc_probs (line 563) | def calculate_all_ctc_probs(self, xs_pad, ilens, ys_pad, ys_pad_src): FILE: nets/pytorch_backend/e2e_tts_fastspeech.py class FeedForwardTransformerLoss (line 34) | class FeedForwardTransformerLoss(torch.nn.Module): method __init__ (line 37) | def __init__(self, use_masking=True, use_weighted_masking=False): method forward (line 57) | def forward(self, after_outs, before_outs, d_outs, ys, ds, ilens, olens): class FeedForwardTransformer (line 112) | class FeedForwardTransformer(TTSInterface, torch.nn.Module): method add_arguments (line 128) | def add_arguments(parser): method __init__ (line 377) | def __init__(self, idim, odim, args=None): method _forward (line 567) | def _forward( method forward (line 627) | def forward(self, xs, ilens, ys, olens, spembs=None, extras=None, *arg... method calculate_all_attentions (line 686) | def calculate_all_attentions( method inference (line 741) | def inference(self, x, inference_args, spemb=None, *args, **kwargs): method _integrate_with_spk_embed (line 777) | def _integrate_with_spk_embed(self, hs, spembs): method _source_mask (line 801) | def _source_mask(self, ilens): method _load_teacher_model (line 822) | def _load_teacher_model(self, model_path): method _reset_parameters (line 844) | def _reset_parameters(self, init_type, init_enc_alpha=1.0, init_dec_al... method _transfer_from_teacher (line 853) | def _transfer_from_teacher(self, transferred_encoder_module): method attention_plot_class (line 874) | def attention_plot_class(self): method base_plot_keys (line 882) | def base_plot_keys(self): FILE: nets/pytorch_backend/e2e_tts_tacotron2.py class GuidedAttentionLoss (line 25) | class GuidedAttentionLoss(torch.nn.Module): method __init__ (line 39) | def __init__(self, sigma=0.4, alpha=1.0, reset_always=True): method _reset_masks (line 56) | def _reset_masks(self): method forward (line 60) | def forward(self, att_ws, ilens, olens): method _make_guided_attention_masks (line 84) | def _make_guided_attention_masks(self, ilens, olens): method _make_guided_attention_mask (line 96) | def _make_guided_attention_mask(ilen, olen, sigma): method _make_masks (line 128) | def _make_masks(ilens, olens): class Tacotron2Loss (line 166) | class Tacotron2Loss(torch.nn.Module): method __init__ (line 169) | def __init__( method forward (line 198) | def forward(self, after_outs, before_outs, logits, ys, labels, olens): method _load_state_dict_pre_hook (line 249) | def _load_state_dict_pre_hook( class Tacotron2 (line 273) | class Tacotron2(TTSInterface, torch.nn.Module): method add_arguments (line 288) | def add_arguments(parser): method __init__ (line 519) | def __init__(self, idim, odim, args=None): method forward (line 702) | def forward( method inference (line 790) | def inference(self, x, inference_args, spemb=None, *args, **kwargs): method calculate_all_attentions (line 838) | def calculate_all_attentions( method base_plot_keys (line 875) | def base_plot_keys(self): FILE: nets/pytorch_backend/e2e_tts_transformer.py class GuidedMultiHeadAttentionLoss (line 31) | class GuidedMultiHeadAttentionLoss(GuidedAttentionLoss): method forward (line 42) | def forward(self, att_ws, ilens, olens): class TTSPlot (line 77) | class TTSPlot(PlotAttentionReport): method plotfn (line 80) | def plotfn( class Transformer (line 115) | class Transformer(TTSInterface, torch.nn.Module): method add_arguments (line 129) | def add_arguments(parser): method attention_plot_class (line 438) | def attention_plot_class(self): method __init__ (line 442) | def __init__(self, idim, odim, args=None): method _reset_parameters (line 677) | def _reset_parameters(self, init_type, init_enc_alpha=1.0, init_dec_al... method _add_first_frame_and_remove_last_frame (line 686) | def _add_first_frame_and_remove_last_frame(self, ys): method forward (line 692) | def forward(self, xs, ilens, ys, labels, olens, spembs=None, *args, **... method inference (line 841) | def inference(self, x, inference_args, spemb=None, *args, **kwargs): method calculate_all_attentions (line 942) | def calculate_all_attentions( method _integrate_with_spk_embed (line 1053) | def _integrate_with_spk_embed(self, hs, spembs): method _source_mask (line 1077) | def _source_mask(self, ilens): method _target_mask (line 1098) | def _target_mask(self, olens): method base_plot_keys (line 1129) | def base_plot_keys(self): FILE: nets/pytorch_backend/e2e_vc_tacotron2.py class Tacotron2 (line 29) | class Tacotron2(TTSInterface, torch.nn.Module): method add_arguments (line 38) | def add_arguments(parser): method __init__ (line 278) | def __init__(self, idim, odim, args=None): method forward (line 491) | def forward( method inference (line 658) | def inference(self, x, inference_args, spemb=None, *args, **kwargs): method calculate_all_attentions (line 706) | def calculate_all_attentions(self, xs, ilens, ys, spembs=None, *args, ... method base_plot_keys (line 752) | def base_plot_keys(self): method _sort_by_length (line 776) | def _sort_by_length(self, xs, ilens): method _revert_sort_by_length (line 780) | def _revert_sort_by_length(self, xs, ilens, sort_idx): FILE: nets/pytorch_backend/e2e_vc_transformer.py class Transformer (line 34) | class Transformer(TTSInterface, torch.nn.Module): method add_arguments (line 53) | def add_arguments(parser): method attention_plot_class (line 372) | def attention_plot_class(self): method __init__ (line 376) | def __init__(self, idim, odim, args=None): method _reset_parameters (line 619) | def _reset_parameters(self, init_type, init_enc_alpha=1.0, init_dec_al... method _add_first_frame_and_remove_last_frame (line 628) | def _add_first_frame_and_remove_last_frame(self, ys): method forward (line 634) | def forward(self, xs, ilens, ys, labels, olens, spembs=None, *args, **... method inference (line 814) | def inference(self, x, inference_args, spemb=None, *args, **kwargs): method calculate_all_attentions (line 928) | def calculate_all_attentions( method _integrate_with_spk_embed (line 1055) | def _integrate_with_spk_embed(self, hs, spembs): method _source_mask (line 1079) | def _source_mask(self, ilens): method _target_mask (line 1100) | def _target_mask(self, olens): method base_plot_keys (line 1131) | def base_plot_keys(self): FILE: nets/pytorch_backend/fastspeech/duration_calculator.py class DurationCalculator (line 16) | class DurationCalculator(torch.nn.Module): method __init__ (line 24) | def __init__(self, teacher_model): method forward (line 44) | def forward(self, xs, ilens, ys, olens, spembs=None): method _calculate_duration (line 82) | def _calculate_duration(att_w, ilen, olen): method _init_diagonal_head (line 87) | def _init_diagonal_head(self, att_ws): method _calculate_encoder_decoder_attentions (line 91) | def _calculate_encoder_decoder_attentions(self, xs, ilens, ys, olens, ... FILE: nets/pytorch_backend/fastspeech/duration_predictor.py class DurationPredictor (line 14) | class DurationPredictor(torch.nn.Module): method __init__ (line 33) | def __init__( method _forward (line 68) | def _forward(self, xs, x_masks=None, is_inference=False): method forward (line 87) | def forward(self, xs, x_masks=None): method inference (line 101) | def inference(self, xs, x_masks=None): class DurationPredictorLoss (line 116) | class DurationPredictorLoss(torch.nn.Module): method __init__ (line 123) | def __init__(self, offset=1.0, reduction="mean"): method forward (line 135) | def forward(self, outputs, targets): FILE: nets/pytorch_backend/fastspeech/length_regulator.py class LengthRegulator (line 20) | class LengthRegulator(torch.nn.Module): method __init__ (line 34) | def __init__(self, pad_value=0.0): method forward (line 48) | def forward(self, xs, ds, alpha=1.0): method _repeat_one_sequence (line 76) | def _repeat_one_sequence(self, x, d): method _legacy_repeat_one_sequence (line 80) | def _legacy_repeat_one_sequence(self, x, d): FILE: nets/pytorch_backend/frontends/beamformer.py function get_power_spectral_density_matrix (line 6) | def get_power_spectral_density_matrix( function get_mvdr_vector (line 40) | def get_mvdr_vector( function apply_beamforming_vector (line 79) | def apply_beamforming_vector( FILE: nets/pytorch_backend/frontends/dnn_beamformer.py class DNN_Beamformer (line 19) | class DNN_Beamformer(torch.nn.Module): method __init__ (line 28) | def __init__( method forward (line 56) | def forward( class AttentionReference (line 139) | class AttentionReference(torch.nn.Module): method __init__ (line 140) | def __init__(self, bidim, att_dim): method forward (line 145) | def forward( FILE: nets/pytorch_backend/frontends/dnn_wpe.py class DNN_WPE (line 11) | class DNN_WPE(torch.nn.Module): method __init__ (line 12) | def __init__( method forward (line 41) | def forward( FILE: nets/pytorch_backend/frontends/feature_transform.py class FeatureTransform (line 13) | class FeatureTransform(torch.nn.Module): method __init__ (line 14) | def __init__( method forward (line 45) | def forward( class LogMel (line 78) | class LogMel(torch.nn.Module): method __init__ (line 98) | def __init__( method extra_repr (line 120) | def extra_repr(self): method forward (line 123) | def forward( class GlobalMVN (line 135) | class GlobalMVN(torch.nn.Module): method __init__ (line 149) | def __init__( method extra_repr (line 174) | def extra_repr(self): method forward (line 180) | def forward( class UtteranceMVN (line 193) | class UtteranceMVN(torch.nn.Module): method __init__ (line 194) | def __init__( method extra_repr (line 202) | def extra_repr(self): method forward (line 205) | def forward( function utterance_mvn (line 213) | def utterance_mvn( function feature_transform_for (line 250) | def feature_transform_for(args, n_fft): FILE: nets/pytorch_backend/frontends/frontend.py class Frontend (line 15) | class Frontend(nn.Module): method __init__ (line 16) | def __init__( method forward (line 88) | def forward( function frontend_for (line 128) | def frontend_for(args, idim): FILE: nets/pytorch_backend/frontends/mask_estimator.py class MaskEstimator (line 13) | class MaskEstimator(torch.nn.Module): method __init__ (line 14) | def __init__(self, type, idim, layers, units, projs, dropout, nmask=1): method forward (line 30) | def forward( FILE: nets/pytorch_backend/gtn_ctc.py class GTNCTCLossFunction (line 10) | class GTNCTCLossFunction(torch.autograd.Function): method create_ctc_graph (line 17) | def create_ctc_graph(target, blank_idx): method forward (line 41) | def forward(ctx, log_probs, targets, blank_idx=0, reduction="none"): method backward (line 87) | def backward(ctx, grad_output): FILE: nets/pytorch_backend/initialization.py function lecun_normal_init_parameters (line 11) | def lecun_normal_init_parameters(module): function uniform_init_parameters (line 34) | def uniform_init_parameters(module): function set_forget_bias_to_one (line 51) | def set_forget_bias_to_one(bias): FILE: nets/pytorch_backend/lm/default.py class DefaultRNNLM (line 18) | class DefaultRNNLM(BatchScorerInterface, LMInterface, nn.Module): method add_arguments (line 29) | def add_arguments(parser): method __init__ (line 69) | def __init__(self, n_vocab, args): method state_dict (line 100) | def state_dict(self): method load_state_dict (line 104) | def load_state_dict(self, d): method forward (line 108) | def forward(self, x, t): method score (line 140) | def score(self, y, state, x): method final_score (line 157) | def final_score(self, state): method batch_score (line 170) | def batch_score( class ClassifierWithState (line 218) | class ClassifierWithState(nn.Module): method __init__ (line 221) | def __init__( method forward (line 240) | def forward(self, state, *args, **kwargs): method predict (line 282) | def predict(self, state, x): method buff_predict (line 296) | def buff_predict(self, state, x, n): method final (line 311) | def final(self, state, index=None): class RNNLM (line 328) | class RNNLM(nn.Module): method __init__ (line 331) | def __init__( method zero_state (line 393) | def zero_state(self, batchsize): method forward (line 398) | def forward(self, state, x): FILE: nets/pytorch_backend/lm/seq_rnn.py class SequentialRNNLM (line 10) | class SequentialRNNLM(LMInterface, torch.nn.Module): method add_arguments (line 19) | def add_arguments(parser): method __init__ (line 40) | def __init__(self, n_vocab, args): method _setup (line 58) | def _setup( method _init_weights (line 98) | def _init_weights(self): method forward (line 108) | def forward(self, x, t): method _before_loss (line 134) | def _before_loss(self, input, hidden): method init_state (line 143) | def init_state(self, x): method score (line 162) | def score(self, y, state, x): FILE: nets/pytorch_backend/lm/transformer.py class TransformerLM (line 20) | class TransformerLM(nn.Module, LMInterface, BatchScorerInterface): method add_arguments (line 24) | def add_arguments(parser): method __init__ (line 79) | def __init__(self, n_vocab, args): method _target_mask (line 137) | def _target_mask(self, ys_in_pad): method forward (line 142) | def forward( method score (line 178) | def score( method score_partial (line 207) | def score_partial( method select_state (line 214) | def select_state(self, states, i): method batch_score (line 218) | def batch_score( FILE: nets/pytorch_backend/maskctc/add_mask_token.py function mask_uniform (line 13) | def mask_uniform(ys_pad, mask_token, eos, ignore_id): FILE: nets/pytorch_backend/maskctc/mask.py function square_mask (line 11) | def square_mask(ys_in_pad, ignore_id): FILE: nets/pytorch_backend/nets_utils.py function to_device (line 12) | def to_device(m, x): function pad_list (line 34) | def pad_list(xs, pad_value): function make_pad_mask (line 64) | def make_pad_mask(lengths, xs=None, length_dim=-1): function make_non_pad_mask (line 179) | def make_non_pad_mask(lengths, xs=None, length_dim=-1): function mask_by_length (line 268) | def mask_by_length(xs, lengths, fill=0): function th_accuracy (line 299) | def th_accuracy(pad_outputs, pad_targets, ignore_label): function to_torch_tensor (line 322) | def to_torch_tensor(x): function get_subsample (line 390) | def get_subsample(train_args, mode, arch): function rename_state_dict (line 471) | def rename_state_dict( function get_activation (line 485) | def get_activation(act): FILE: nets/pytorch_backend/rnn/argument.py function add_arguments_rnn_encoder_common (line 7) | def add_arguments_rnn_encoder_common(group): function add_arguments_rnn_decoder_common (line 60) | def add_arguments_rnn_decoder_common(group): function add_arguments_rnn_attention_common (line 99) | def add_arguments_rnn_attention_common(group): FILE: nets/pytorch_backend/rnn/attentions.py function _apply_attention_constraint (line 13) | def _apply_attention_constraint( class NoAtt (line 46) | class NoAtt(torch.nn.Module): method __init__ (line 49) | def __init__(self): method reset (line 56) | def reset(self): method forward (line 63) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev): class AttDot (line 94) | class AttDot(torch.nn.Module): method __init__ (line 104) | def __init__(self, eprojs, dunits, att_dim, han_mode=False): method reset (line 118) | def reset(self): method forward (line 125) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev, scaling=2.0): class AttAdd (line 171) | class AttAdd(torch.nn.Module): method __init__ (line 181) | def __init__(self, eprojs, dunits, att_dim, han_mode=False): method reset (line 195) | def reset(self): method forward (line 202) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev, scaling=2.0): class AttLoc (line 250) | class AttLoc(torch.nn.Module): method __init__ (line 265) | def __init__( method reset (line 290) | def reset(self): method forward (line 297) | def forward( class AttCov (line 383) | class AttCov(torch.nn.Module): method __init__ (line 396) | def __init__(self, eprojs, dunits, att_dim, han_mode=False): method reset (line 412) | def reset(self): method forward (line 419) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev_list, scalin... class AttLoc2D (line 485) | class AttLoc2D(torch.nn.Module): method __init__ (line 502) | def __init__( method reset (line 529) | def reset(self): method forward (line 536) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev, scaling=2.0): class AttLocRec (line 606) | class AttLocRec(torch.nn.Module): method __init__ (line 622) | def __init__( method reset (line 647) | def reset(self): method forward (line 654) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev_states, scal... class AttCovLoc (line 729) | class AttCovLoc(torch.nn.Module): method __init__ (line 743) | def __init__( method reset (line 769) | def reset(self): method forward (line 776) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev_list, scalin... class AttMultiHeadDot (line 845) | class AttMultiHeadDot(torch.nn.Module): method __init__ (line 860) | def __init__(self, eprojs, dunits, aheads, att_dim_k, att_dim_v, han_m... method reset (line 883) | def reset(self): method forward (line 891) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev): class AttMultiHeadAdd (line 958) | class AttMultiHeadAdd(torch.nn.Module): method __init__ (line 975) | def __init__(self, eprojs, dunits, aheads, att_dim_k, att_dim_v, han_m... method reset (line 1000) | def reset(self): method forward (line 1008) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev): class AttMultiHeadLoc (line 1075) | class AttMultiHeadLoc(torch.nn.Module): method __init__ (line 1094) | def __init__( method reset (line 1141) | def reset(self): method forward (line 1149) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev, scaling=2.0): class AttMultiHeadMultiResLoc (line 1232) | class AttMultiHeadMultiResLoc(torch.nn.Module): method __init__ (line 1254) | def __init__( method reset (line 1298) | def reset(self): method forward (line 1306) | def forward(self, enc_hs_pad, enc_hs_len, dec_z, att_prev): class AttForward (line 1388) | class AttForward(torch.nn.Module): method __init__ (line 1402) | def __init__(self, eprojs, dunits, att_dim, aconv_chans, aconv_filts): method reset (line 1423) | def reset(self): method forward (line 1430) | def forward( class AttForwardTA (line 1518) | class AttForwardTA(torch.nn.Module): method __init__ (line 1533) | def __init__(self, eunits, dunits, att_dim, aconv_chans, aconv_filts, ... method reset (line 1556) | def reset(self): method forward (line 1563) | def forward( function att_for (line 1661) | def att_for(args, num_att=1, han_mode=False): function initial_att (line 1726) | def initial_att( function att_to_numpy (line 1774) | def att_to_numpy(att_ws, att): FILE: nets/pytorch_backend/rnn/decoders.py class Decoder (line 29) | class Decoder(torch.nn.Module, ScorerInterface): method __init__ (line 50) | def __init__( method zero_state (line 124) | def zero_state(self, hs_pad): method rnn_forward (line 127) | def rnn_forward(self, ey, z_list, c_list, z_prev, c_prev): method forward (line 142) | def forward(self, hs_pad, hlens, ys_pad, strm_idx=0, lang_ids=None): method recognize_beam (line 313) | def recognize_beam(self, h, lpz, recog_args, char_list, rnnlm=None, st... method recognize_beam_batch (line 632) | def recognize_beam_batch( method calculate_all_attentions (line 969) | def calculate_all_attentions(self, hs_pad, hlen, ys_pad, strm_idx=0, l... method _get_last_yseq (line 1080) | def _get_last_yseq(exp_yseq): method _append_ids (line 1087) | def _append_ids(yseq, ids): method _index_select_list (line 1097) | def _index_select_list(yseq, lst): method _index_select_lm_state (line 1104) | def _index_select_lm_state(rnnlm_state, dim, vidx): method init_state (line 1116) | def init_state(self, x): method score (line 1144) | def score(self, yseq, state, x): function decoder_for (line 1199) | def decoder_for(args, odim, sos, eos, att, labeldist): FILE: nets/pytorch_backend/rnn/encoders.py class RNNP (line 15) | class RNNP(torch.nn.Module): method __init__ (line 27) | def __init__(self, idim, elayers, cdim, hdim, subsample, dropout, typ=... method forward (line 56) | def forward(self, xs_pad, ilens, prev_state=None): class RNN (line 95) | class RNN(torch.nn.Module): method __init__ (line 106) | def __init__(self, idim, elayers, cdim, hdim, dropout, typ="blstm"): method forward (line 134) | def forward(self, xs_pad, ilens, prev_state=None): function reset_backward_rnn_state (line 165) | def reset_backward_rnn_state(states): class VGG2L (line 178) | class VGG2L(torch.nn.Module): method __init__ (line 184) | def __init__(self, in_channel=1): method forward (line 194) | def forward(self, xs_pad, ilens, **kwargs): class Encoder (line 240) | class Encoder(torch.nn.Module): method __init__ (line 253) | def __init__( method forward (line 307) | def forward(self, xs_pad, ilens, prev_states=None): function encoder_for (line 331) | def encoder_for(args, idim, subsample): FILE: nets/pytorch_backend/streaming/segment.py class SegmentStreamingE2E (line 5) | class SegmentStreamingE2E(object): method __init__ (line 12) | def __init__(self, e2e, recog_args, rnnlm=None): method accept_input (line 42) | def accept_input(self, x): FILE: nets/pytorch_backend/streaming/window.py class WindowStreamingE2E (line 6) | class WindowStreamingE2E(object): method __init__ (line 13) | def __init__(self, e2e, recog_args, rnnlm=None): method accept_input (line 31) | def accept_input(self, x): method _input_window_for_decoder (line 45) | def _input_window_for_decoder(self, use_all=False): method decode_with_attention_offline (line 68) | def decode_with_attention_offline(self): FILE: nets/pytorch_backend/tacotron2/cbhg.py class CBHGLoss (line 18) | class CBHGLoss(torch.nn.Module): method __init__ (line 21) | def __init__(self, use_masking=True): method forward (line 31) | def forward(self, cbhg_outs, spcs, olens): class CBHG (line 57) | class CBHG(torch.nn.Module): method __init__ (line 69) | def __init__( method forward (line 172) | def forward(self, xs, ilens): method inference (line 216) | def inference(self, x): method _sort_by_length (line 232) | def _sort_by_length(self, xs, ilens): method _revert_sort_by_length (line 236) | def _revert_sort_by_length(self, xs, ilens, sort_idx): class HighwayNet (line 241) | class HighwayNet(torch.nn.Module): method __init__ (line 250) | def __init__(self, idim): method forward (line 264) | def forward(self, x): FILE: nets/pytorch_backend/tacotron2/decoder.py function decoder_init (line 17) | def decoder_init(m): class ZoneOutCell (line 23) | class ZoneOutCell(torch.nn.Module): method __init__ (line 42) | def __init__(self, cell, zoneout_rate=0.1): method forward (line 60) | def forward(self, inputs, hidden): method _zoneout (line 79) | def _zoneout(self, h, next_h, prob): class Prenet (line 96) | class Prenet(torch.nn.Module): method __init__ (line 116) | def __init__(self, idim, n_layers=2, n_units=256, dropout_rate=0.5): method forward (line 135) | def forward(self, x): class Postnet (line 150) | class Postnet(torch.nn.Module): method __init__ (line 165) | def __init__( method forward (line 254) | def forward(self, xs): class Decoder (line 269) | class Decoder(torch.nn.Module): method __init__ (line 283) | def __init__( method _zero_state (line 388) | def _zero_state(self, hs): method forward (line 392) | def forward(self, hs, hlens, ys): method inference (line 482) | def inference( method calculate_all_attentions (line 618) | def calculate_all_attentions(self, hs, hlens, ys): FILE: nets/pytorch_backend/tacotron2/encoder.py function encoder_init (line 17) | def encoder_init(m): class Encoder (line 23) | class Encoder(torch.nn.Module): method __init__ (line 36) | def __init__( method forward (line 130) | def forward(self, xs, ilens=None): method inference (line 162) | def inference(self, x): FILE: nets/pytorch_backend/transducer/arguments.py function add_encoder_general_arguments (line 7) | def add_encoder_general_arguments(group): function add_rnn_encoder_arguments (line 44) | def add_rnn_encoder_arguments(group): function add_custom_encoder_arguments (line 74) | def add_custom_encoder_arguments(group): function add_decoder_general_arguments (line 128) | def add_decoder_general_arguments(group): function add_rnn_decoder_arguments (line 153) | def add_rnn_decoder_arguments(group): function add_custom_decoder_arguments (line 171) | def add_custom_decoder_arguments(group): function add_custom_training_arguments (line 204) | def add_custom_training_arguments(group): function add_transducer_arguments (line 222) | def add_transducer_arguments(group): function add_auxiliary_task_arguments (line 261) | def add_auxiliary_task_arguments(group): function add_att_scorer_arguments (line 368) | def add_att_scorer_arguments(group): function add_transducer_code_switch_arguments (line 453) | def add_transducer_code_switch_arguments(group): FILE: nets/pytorch_backend/transducer/auxiliary_task.py class AuxiliaryTask (line 14) | class AuxiliaryTask(torch.nn.Module): method __init__ (line 17) | def __init__( method forward (line 54) | def forward( FILE: nets/pytorch_backend/transducer/blocks.py function check_and_prepare (line 38) | def check_and_prepare(net_part, blocks_arch, input_layer): function get_pos_enc_and_att_class (line 208) | def get_pos_enc_and_att_class(net_part, pos_enc_type, self_attn_type): function build_input_layer (line 242) | def build_input_layer( function build_transformer_block (line 311) | def build_transformer_block(net_part, block_arch, pw_layer_type, pw_acti... function build_conformer_block (line 356) | def build_conformer_block( function build_causal_conv1d_block (line 412) | def build_causal_conv1d_block(block_arch): function build_tdnn_block (line 429) | def build_tdnn_block(block_arch): function build_blocks (line 464) | def build_blocks( FILE: nets/pytorch_backend/transducer/causal_conv1d.py class CausalConv1d (line 6) | class CausalConv1d(torch.nn.Module): method __init__ (line 20) | def __init__( method forward (line 39) | def forward(self, x, x_mask, cache=None): FILE: nets/pytorch_backend/transducer/custom_decoder.py class CustomDecoder (line 14) | class CustomDecoder(TransducerDecoderInterface, torch.nn.Module): method __init__ (line 30) | def __init__( method set_device (line 67) | def set_device(self, device): method init_state (line 76) | def init_state(self, batch_size=None, device=None, dtype=None): method forward (line 90) | def forward(self, tgt, tgt_mask, memory): method score (line 115) | def score(self, hyp, cache): method batch_score (line 154) | def batch_score(self, hyps, batch_states, cache, use_lm): method select_state (line 232) | def select_state(self, batch_states, idx): method create_batch_states (line 252) | def create_batch_states(self, batch_states, l_states, check_list): FILE: nets/pytorch_backend/transducer/custom_encoder.py class CustomEncoder (line 12) | class CustomEncoder(torch.nn.Module): method __init__ (line 31) | def __init__( method forward (line 76) | def forward(self, xs, masks, return_as_intermidiate=False): FILE: nets/pytorch_backend/transducer/error_calculator.py class ErrorCalculator (line 11) | class ErrorCalculator(object): method __init__ (line 25) | def __init__( method __call__ (line 53) | def __call__(self, hs_pad, ys_pad): method convert_to_char (line 88) | def convert_to_char(self, ys_hat, ys_pad): method calculate_cer (line 117) | def calculate_cer(self, seqs_hat, seqs_true): method calculate_wer (line 141) | def calculate_wer(self, seqs_hat, seqs_true): FILE: nets/pytorch_backend/transducer/initializer.py function initializer (line 11) | def initializer(model, args): FILE: nets/pytorch_backend/transducer/joint_network.py class JointNetwork (line 8) | class JointNetwork(torch.nn.Module): method __init__ (line 17) | def __init__( method forward (line 37) | def forward( FILE: nets/pytorch_backend/transducer/loss.py class TransLoss (line 8) | class TransLoss(torch.nn.Module): method __init__ (line 16) | def __init__(self, trans_type, blank_id): method forward (line 43) | def forward(self, pred_pad, target, pred_len, target_len): FILE: nets/pytorch_backend/transducer/rnn_decoder.py class DecoderRNNT (line 8) | class DecoderRNNT(TransducerDecoderInterface, torch.nn.Module): method __init__ (line 23) | def __init__( method set_device (line 61) | def set_device(self, device): method set_data_type (line 70) | def set_data_type(self, data_type): method init_state (line 79) | def init_state(self, batch_size): method rnn_forward (line 111) | def rnn_forward(self, y, state): method forward (line 145) | def forward(self, hs_pad, ys_in_pad): method score (line 168) | def score(self, hyp, cache): method batch_score (line 196) | def batch_score(self, hyps, batch_states, cache, use_lm): method select_state (line 255) | def select_state(self, batch_states, idx): method create_batch_states (line 273) | def create_batch_states(self, batch_states, l_states, l_tokens=None): FILE: nets/pytorch_backend/transducer/rnn_encoder.py class RNNP (line 30) | class RNNP(torch.nn.Module): method __init__ (line 45) | def __init__( method forward (line 87) | def forward( class RNN (line 150) | class RNN(torch.nn.Module): method __init__ (line 163) | def __init__( method forward (line 203) | def forward( function reset_backward_rnn_state (line 271) | def reset_backward_rnn_state( class VGG2L (line 291) | class VGG2L(torch.nn.Module): method __init__ (line 299) | def __init__(self, in_channel: int = 1): method forward (line 311) | def forward(self, xs_pad: torch.Tensor, ilens: torch.Tensor, **kwargs): class Encoder (line 357) | class Encoder(torch.nn.Module): method __init__ (line 372) | def __init__( method forward (line 460) | def forward(self, xs_pad, ilens, prev_states=None): function encoder_for (line 509) | def encoder_for( FILE: nets/pytorch_backend/transducer/tdnn.py class TDNN (line 10) | class TDNN(torch.nn.Module): method __init__ (line 25) | def __init__( method forward (line 61) | def forward( method create_outputs (line 108) | def create_outputs( FILE: nets/pytorch_backend/transducer/transformer_decoder_layer.py class DecoderLayer (line 9) | class DecoderLayer(nn.Module): method __init__ (line 21) | def __init__(self, size, self_attn, feed_forward, dropout_rate): method forward (line 35) | def forward(self, tgt, tgt_mask, cache=None): FILE: nets/pytorch_backend/transducer/utils.py function prepare_loss_inputs (line 11) | def prepare_loss_inputs(ys_pad, hlens, blank_id=0, ignore_id=-1): function valid_aux_task_layer_list (line 51) | def valid_aux_task_layer_list(aux_layer_ids, enc_num_layers): function is_prefix (line 83) | def is_prefix(x, pref): function substract (line 104) | def substract(x, subset): function select_lm_state (line 125) | def select_lm_state(lm_states, idx, lm_layers, is_wordlm): function create_lm_batch_state (line 149) | def create_lm_batch_state(lm_states_list, lm_layers, is_wordlm): function init_lm_state (line 178) | def init_lm_state(lm_model): function recombine_hyps (line 210) | def recombine_hyps(hyps, mmi_weight): function pad_sequence (line 241) | def pad_sequence(seqlist, pad_token): function check_state (line 259) | def check_state(state, max_len, pad_token): function check_batch_state (line 296) | def check_batch_state(state, max_len, pad_token): function custom_torch_load (line 322) | def custom_torch_load(model_path, model, training=True): FILE: nets/pytorch_backend/transducer/vgg2l.py class VGG2L (line 9) | class VGG2L(torch.nn.Module): method __init__ (line 19) | def __init__(self, idim: int, odim: int, pos_enc: torch.nn.Module = No... method forward (line 43) | def forward( method create_new_mask (line 71) | def create_new_mask(self, x_mask: torch.Tensor) -> torch.Tensor: FILE: nets/pytorch_backend/transformer/add_sos_eos.py function add_sos_eos (line 12) | def add_sos_eos(ys_pad, sos, eos, ignore_id): FILE: nets/pytorch_backend/transformer/argument.py function add_arguments_transformer_common (line 10) | def add_arguments_transformer_common(group): FILE: nets/pytorch_backend/transformer/attention.py class MultiHeadedAttention (line 16) | class MultiHeadedAttention(nn.Module): method __init__ (line 26) | def __init__(self, n_head, n_feat, dropout_rate): method forward_qkv (line 40) | def forward_qkv(self, query, key, value): method forward_attention (line 64) | def forward_attention(self, value, scores, mask): method forward (line 98) | def forward(self, query, key, value, mask): class LegacyRelPositionMultiHeadedAttention (line 117) | class LegacyRelPositionMultiHeadedAttention(MultiHeadedAttention): method __init__ (line 132) | def __init__(self, n_head, n_feat, dropout_rate, zero_triu=False): method rel_shift (line 145) | def rel_shift(self, x): method forward (line 167) | def forward(self, query, key, value, pos_emb, mask): class RelPositionMultiHeadedAttention (line 212) | class RelPositionMultiHeadedAttention(MultiHeadedAttention): method __init__ (line 227) | def __init__(self, n_head, n_feat, dropout_rate, zero_triu=False): method rel_shift (line 240) | def rel_shift(self, x): method forward (line 265) | def forward(self, query, key, value, pos_emb, mask): FILE: nets/pytorch_backend/transformer/contextual_block_encoder_layer.py class ContextualBlockEncoderLayer (line 16) | class ContextualBlockEncoderLayer(nn.Module): method __init__ (line 37) | def __init__( method forward (line 61) | def forward(self, x, mask, past_ctx=None, next_ctx=None, layer_idx=0, ... FILE: nets/pytorch_backend/transformer/decoder.py function _pre_hook (line 34) | def _pre_hook( class Decoder (line 47) | class Decoder(BatchScorerInterface, torch.nn.Module): method __init__ (line 80) | def __init__( method forward (line 220) | def forward(self, tgt, tgt_mask, memory, memory_mask): method forward_one_step (line 252) | def forward_one_step(self, tgt, tgt_mask, memory, cache=None): method score (line 289) | def score(self, ys, state, x): method batch_score (line 304) | def batch_score( FILE: nets/pytorch_backend/transformer/decoder_layer.py class DecoderLayer (line 15) | class DecoderLayer(nn.Module): method __init__ (line 37) | def __init__( method forward (line 63) | def forward(self, tgt, tgt_mask, memory, memory_mask, cache=None): FILE: nets/pytorch_backend/transformer/dynamic_conv.py class DynamicConvolution (line 12) | class DynamicConvolution(nn.Module): method __init__ (line 28) | def __init__( method forward (line 61) | def forward(self, query, key, value, mask): FILE: nets/pytorch_backend/transformer/dynamic_conv2d.py class DynamicConvolution2D (line 12) | class DynamicConvolution2D(nn.Module): method __init__ (line 28) | def __init__( method forward (line 65) | def forward(self, query, key, value, mask): FILE: nets/pytorch_backend/transformer/embedding.py function _pre_hook (line 14) | def _pre_hook( class PositionalEncoding (line 35) | class PositionalEncoding(torch.nn.Module): method __init__ (line 48) | def __init__(self, d_model, dropout_rate, max_len=5000, reverse=False): method extend_pe (line 59) | def extend_pe(self, x): method forward (line 82) | def forward(self, x: torch.Tensor): class ScaledPositionalEncoding (line 97) | class ScaledPositionalEncoding(PositionalEncoding): method __init__ (line 109) | def __init__(self, d_model, dropout_rate, max_len=5000): method reset_parameters (line 114) | def reset_parameters(self): method forward (line 118) | def forward(self, x): class LegacyRelPositionalEncoding (line 133) | class LegacyRelPositionalEncoding(PositionalEncoding): method __init__ (line 147) | def __init__(self, d_model, dropout_rate, max_len=5000): method forward (line 156) | def forward(self, x): class RelPositionalEncoding (line 173) | class RelPositionalEncoding(torch.nn.Module): method __init__ (line 187) | def __init__(self, d_model, dropout_rate, max_len=5000): method extend_pe (line 196) | def extend_pe(self, x): method forward (line 228) | def forward(self, x: torch.Tensor): FILE: nets/pytorch_backend/transformer/encoder.py function _pre_hook (line 30) | def _pre_hook( class Encoder (line 45) | class Encoder(torch.nn.Module): method __init__ (line 79) | def __init__( method get_positionwise_layer (line 250) | def get_positionwise_layer( method forward (line 282) | def forward(self, xs, masks): method forward_one_step (line 306) | def forward_one_step(self, xs, masks, cache=None): FILE: nets/pytorch_backend/transformer/encoder_layer.py class EncoderLayer (line 16) | class EncoderLayer(nn.Module): method __init__ (line 36) | def __init__( method forward (line 58) | def forward(self, x, mask, cache=None): FILE: nets/pytorch_backend/transformer/encoder_mix.py class EncoderMix (line 20) | class EncoderMix(Encoder, torch.nn.Module): method __init__ (line 43) | def __init__( method forward (line 109) | def forward(self, xs, masks): method forward_one_step (line 130) | def forward_one_step(self, xs, masks, cache=None): FILE: nets/pytorch_backend/transformer/initializer.py function initialize (line 14) | def initialize(model, init_type="pytorch"): FILE: nets/pytorch_backend/transformer/label_smoothing_loss.py class LabelSmoothingLoss (line 13) | class LabelSmoothingLoss(nn.Module): method __init__ (line 23) | def __init__( method forward (line 41) | def forward(self, x, target): FILE: nets/pytorch_backend/transformer/layer_norm.py class LayerNorm (line 12) | class LayerNorm(torch.nn.LayerNorm): method __init__ (line 21) | def __init__(self, nout, dim=-1): method forward (line 26) | def forward(self, x): FILE: nets/pytorch_backend/transformer/lightconv.py class LightweightConvolution (line 12) | class LightweightConvolution(nn.Module): method __init__ (line 28) | def __init__( method forward (line 65) | def forward(self, query, key, value, mask): FILE: nets/pytorch_backend/transformer/lightconv2d.py class LightweightConvolution2D (line 12) | class LightweightConvolution2D(nn.Module): method __init__ (line 28) | def __init__( method forward (line 66) | def forward(self, query, key, value, mask): FILE: nets/pytorch_backend/transformer/mask.py function subsequent_mask (line 20) | def subsequent_mask(size, device="cpu", dtype=datatype): function target_mask (line 41) | def target_mask(ys_in_pad, ignore_id): FILE: nets/pytorch_backend/transformer/multi_layer_conv.py class MultiLayeredConv1d (line 12) | class MultiLayeredConv1d(torch.nn.Module): method __init__ (line 25) | def __init__(self, in_chans, hidden_chans, kernel_size, dropout_rate): method forward (line 52) | def forward(self, x): class Conv1dLinear (line 66) | class Conv1dLinear(torch.nn.Module): method __init__ (line 73) | def __init__(self, in_chans, hidden_chans, kernel_size, dropout_rate): method forward (line 94) | def forward(self, x): FILE: nets/pytorch_backend/transformer/optimizer.py class NoamOpt (line 12) | class NoamOpt(object): method __init__ (line 15) | def __init__(self, model_size, factor, warmup, optimizer): method param_groups (line 25) | def param_groups(self): method step (line 29) | def step(self): method rate (line 38) | def rate(self, step=None): method zero_grad (line 48) | def zero_grad(self): method state_dict (line 52) | def state_dict(self): method load_state_dict (line 63) | def load_state_dict(self, state_dict): function get_std_opt (line 72) | def get_std_opt(model_params, d_model, warmup, factor): FILE: nets/pytorch_backend/transformer/plot.py function _plot_and_save_attention (line 13) | def _plot_and_save_attention(att_w, filename, xtokens=None, ytokens=None): function savefig (line 45) | def savefig(plot, filename): function plot_multi_head_attention (line 50) | def plot_multi_head_attention( class PlotAttentionReport (line 117) | class PlotAttentionReport(asr_utils.PlotAttentionReport): method plotfn (line 118) | def plotfn(self, *args, **kwargs): method __call__ (line 126) | def __call__(self, trainer): method get_attention_weights (line 131) | def get_attention_weights(self): method log_attentions (line 140) | def log_attentions(self, logger, step): FILE: nets/pytorch_backend/transformer/positionwise_feed_forward.py class PositionwiseFeedForward (line 12) | class PositionwiseFeedForward(torch.nn.Module): method __init__ (line 22) | def __init__(self, idim, hidden_units, dropout_rate, activation=torch.... method forward (line 30) | def forward(self, x): FILE: nets/pytorch_backend/transformer/repeat.py class MultiSequential (line 12) | class MultiSequential(torch.nn.Sequential): method forward (line 15) | def forward(self, *args): function repeat (line 22) | def repeat(N, fn): FILE: nets/pytorch_backend/transformer/sgd_optimizer.py class NoamOpt (line 12) | class NoamOpt(object): method __init__ (line 15) | def __init__(self, model_size, factor, warmup, optimizer): method param_groups (line 25) | def param_groups(self): method step (line 29) | def step(self): method rate (line 38) | def rate(self, step=None): method zero_grad (line 48) | def zero_grad(self): method state_dict (line 52) | def state_dict(self): method load_state_dict (line 63) | def load_state_dict(self, state_dict): function get_sgd_opt (line 72) | def get_sgd_opt(model_params, d_model, warmup, factor): FILE: nets/pytorch_backend/transformer/subsampling.py class TooShortUttError (line 14) | class TooShortUttError(Exception): method __init__ (line 24) | def __init__(self, message, actual_size, limit): function check_short_utt (line 31) | def check_short_utt(ins, size): class Conv2dSubsampling (line 42) | class Conv2dSubsampling(torch.nn.Module): method __init__ (line 53) | def __init__(self, idim, odim, dropout_rate, pos_enc=None): method forward (line 67) | def forward(self, x, x_mask): method __getitem__ (line 89) | def __getitem__(self, key): class Conv2dSubsampling6 (line 101) | class Conv2dSubsampling6(torch.nn.Module): method __init__ (line 112) | def __init__(self, idim, odim, dropout_rate, pos_enc=None): method forward (line 126) | def forward(self, x, x_mask): class Conv2dSubsampling8 (line 149) | class Conv2dSubsampling8(torch.nn.Module): method __init__ (line 160) | def __init__(self, idim, odim, dropout_rate, pos_enc=None): method forward (line 176) | def forward(self, x, x_mask): FILE: nets/pytorch_backend/transformer/subsampling_without_posenc.py class Conv2dSubsamplingWOPosEnc (line 10) | class Conv2dSubsamplingWOPosEnc(torch.nn.Module): method __init__ (line 22) | def __init__(self, idim, odim, dropout_rate, kernels, strides): method forward (line 39) | def forward(self, x, x_mask): FILE: nets/pytorch_backend/wavenet.py function encode_mu_law (line 19) | def encode_mu_law(x, mu=256): function decode_mu_law (line 35) | def decode_mu_law(y, mu=256): function initialize (line 52) | def initialize(m): class OneHot (line 68) | class OneHot(nn.Module): method __init__ (line 76) | def __init__(self, depth): method forward (line 80) | def forward(self, x): class CausalConv1d (line 97) | class CausalConv1d(nn.Module): method __init__ (line 100) | def __init__(self, in_channels, out_channels, kernel_size, dilation=1,... method forward (line 116) | def forward(self, x): class UpSampling (line 132) | class UpSampling(nn.Module): method __init__ (line 140) | def __init__(self, upsampling_factor, bias=True): method forward (line 152) | def forward(self, x): class WaveNet (line 167) | class WaveNet(nn.Module): method __init__ (line 183) | def __init__( method forward (line 238) | def forward(self, x, h): method generate (line 275) | def generate(self, x, h, n_samples, interval=None, mode="sampling"): method _preprocess (line 392) | def _preprocess(self, x): method _postprocess (line 397) | def _postprocess(self, x): method _residual_forward (line 404) | def _residual_forward( method _generate_residual_forward (line 427) | def _generate_residual_forward( FILE: nets/scorer_interface.py class ScorerInterface (line 11) | class ScorerInterface: method init_state (line 29) | def init_state(self, x: torch.Tensor) -> Any: method select_state (line 40) | def select_state(self, state: Any, i: int, new_id: int = None) -> Any: method score (line 54) | def score( method final_score (line 72) | def final_score(self, state: Any) -> float: class BatchScorerInterface (line 85) | class BatchScorerInterface(ScorerInterface): method batch_init_state (line 88) | def batch_init_state(self, x: torch.Tensor) -> Any: method batch_score (line 99) | def batch_score( class PartialScorerInterface (line 131) | class PartialScorerInterface(ScorerInterface): method score_partial (line 144) | def score_partial( class BatchPartialScorerInterface (line 164) | class BatchPartialScorerInterface(BatchScorerInterface, PartialScorerInt... method batch_score_partial (line 167) | def batch_score_partial( FILE: nets/scorers/_mmi_utils.py function intersect_dense_forward (line 26) | def intersect_dense_forward(a_fsas: Fsa, function step_trace (line 79) | def step_trace(out_fsas, a_fsas, arc_map_a): function step_intersect (line 170) | def step_intersect(fsa, function step_intersect_test (line 192) | def step_intersect_test(): FILE: nets/scorers/ctc.py class CTCPrefixScorer (line 11) | class CTCPrefixScorer(BatchPartialScorerInterface): method __init__ (line 14) | def __init__(self, ctc: torch.nn.Module, eos: int): method init_state (line 27) | def init_state(self, x: torch.Tensor): method select_state (line 41) | def select_state(self, state, i, new_id=None): method score_partial (line 66) | def score_partial(self, y, ids, state, x): method batch_init_state (line 88) | def batch_init_state(self, x: torch.Tensor): method batch_score_partial (line 102) | def batch_score_partial(self, y, ids, state, x): method extend_prob (line 129) | def extend_prob(self, x: torch.Tensor): method extend_state (line 142) | def extend_state(self, state): FILE: nets/scorers/ctc_rnnt_scorer.py class CTCRNNTScorer (line 8) | class CTCRNNTScorer(): method __init__ (line 9) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 29) | def load_weight(self, rank): method den_scores (line 38) | def den_scores(self, x): method batch_score (line 47) | def batch_score(self, A, nnet_output, den_scores, tu_sum, mmi_weight): FILE: nets/scorers/length_bonus.py class LengthBonus (line 11) | class LengthBonus(BatchScorerInterface): method __init__ (line 14) | def __init__(self, n_vocab: int): method score (line 23) | def score(self, y, state, x): method batch_score (line 39) | def batch_score( FILE: nets/scorers/lookahead.py function search_lexical_tree (line 5) | def search_lexical_tree(self, node, next_tokens): function parse_lookahead (line 70) | def parse_lookahead(yseq, lexroot, char_list, alphabet, word_dict, bpe_s... function build_word_fsa_mat (line 116) | def build_word_fsa_mat(prefix, interval): FILE: nets/scorers/mmi.py class MMIPrefixScores (line 13) | class MMIPrefixScores(PartialScorerInterface): method __init__ (line 14) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 43) | def load_weight(self, rank): method init_state (line 53) | def init_state(self, x): method select_state (line 75) | def select_state(self, states, j): method score (line 80) | def score(**kargs): method score_partial (line 83) | def score_partial(self, y, next_tokens, state, hs_pad): method score (line 127) | def score(self, state): method final_score (line 130) | def final_score(self, state): method segmentation (line 135) | def segmentation(self, ys): FILE: nets/scorers/mmi_alignment_score.py class MMIRNNTScorer (line 10) | class MMIRNNTScorer(): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 34) | def load_weight(self, rank): method den_scores (line 44) | def den_scores(self, x): method batch_rescore (line 60) | def batch_rescore(self, A, nnet_output, den_scores): method batch_score (line 88) | def batch_score(self, A, nnet_output, den_scores, tu_sum, mmi_weight): method Imp_B (line 94) | def Imp_B(self, A, nnet_output, den_scores, tu_sum, mmi_weight): method Imp_A (line 156) | def Imp_A(self, A, nnet_output, den_scores, tu_sum, mmi_weight): FILE: nets/scorers/mmi_frame_prefix_scorer.py class MMIFramePrefixScorer (line 10) | class MMIFramePrefixScorer(PartialScorerInterface): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 45) | def load_weight(self, rank): method init_state (line 55) | def init_state(self, x): method select_state (line 81) | def select_state(self, states, j): method score (line 85) | def score(**kargs): method split_intervals (line 88) | def split_intervals(self, olds, max_interval=2000): method score_partial (line 102) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 188) | def final_score(self, state): FILE: nets/scorers/mmi_frame_scorer.py class MMIFrameScorer (line 11) | class MMIFrameScorer(PartialScorerInterface): method __init__ (line 12) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 39) | def load_weight(self, rank): method init_state (line 49) | def init_state(self, x): method select_state (line 90) | def select_state(self, states, j): method score (line 94) | def score(**kargs): method score_partial (line 97) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 180) | def final_score(self, state): FILE: nets/scorers/mmi_frame_scorer_trace.py class MMIFrameScorer (line 9) | class MMIFrameScorer(PartialScorerInterface): method __init__ (line 10) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 31) | def load_weight(self, rank): method init_state (line 41) | def init_state(self, x): method select_state (line 75) | def select_state(self, states, j): method score (line 79) | def score(**kargs): method score_partial (line 82) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 172) | def final_score(self, state): FILE: nets/scorers/mmi_lookahead.py class MMILookaheadScorer (line 10) | class MMILookaheadScorer(PartialScorerInterface): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 44) | def load_weight(self, rank): method init_state (line 54) | def init_state(self, x): method select_state (line 78) | def select_state(self, states, j): method score (line 83) | def score(**kargs): method score_partial (line 86) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 210) | def final_score(self, state): FILE: nets/scorers/mmi_lookahead_bak.py class MMILookaheadScorer (line 10) | class MMILookaheadScorer(PartialScorerInterface): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 38) | def load_weight(self, rank): method init_state (line 48) | def init_state(self, x): method select_state (line 59) | def select_state(self, states, j): method score (line 63) | def score(**kargs): method score_partial (line 66) | def score_partial(self, y, next_tokens, state, hs_pad): method split_intervals (line 123) | def split_intervals(self, olds, max_interval=2000): FILE: nets/scorers/mmi_lookahead_split.py class MMILookaheadScorer (line 10) | class MMILookaheadScorer(PartialScorerInterface): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 42) | def load_weight(self, rank): method init_state (line 52) | def init_state(self, x): method select_state (line 76) | def select_state(self, states, j): method score (line 81) | def score(**kargs): method score_partial (line 84) | def score_partial(self, y, next_tokens, state, hs_pad): method search_lexical_tree (line 162) | def search_lexical_tree(self, node, next_tokens): method split_intervals (line 227) | def split_intervals(self, intervals, is_new_word, max_len=20000): method final_score (line 247) | def final_score(self, state): FILE: nets/scorers/mmi_prefix_score.py class MMIFrameScorer (line 10) | class MMIFrameScorer(PartialScorerInterface): method __init__ (line 11) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 38) | def load_weight(self, rank, path): method init_state (line 49) | def init_state(self, x): method select_state (line 66) | def select_state(self, states, j): method score (line 70) | def score(**kargs): method score_partial (line 73) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 114) | def final_score(self, state): FILE: nets/scorers/mmi_rescorer.py class MMIRescorer (line 9) | class MMIRescorer(object): method __init__ (line 10) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 35) | def load_weight(self, rank, path): method score (line 46) | def score(self, x, nbest_hyps, v2=False): FILE: nets/scorers/mmi_rnnt_lookahead_scorer.py class MMIRNNTLookaheadScorer (line 12) | class MMIRNNTLookaheadScorer(): method __init__ (line 13) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 45) | def load_weight(self, rank): method den_scores (line 55) | def den_scores(self, x): method batch_score_ (line 79) | def batch_score_(self, A, nnet_output, den_scores, tu_sum, mmi_weight): method __batch_score (line 133) | def __batch_score(self, A, nnet_output, den_scores, tu_sum, mmi_weight): method batch_score (line 193) | def batch_score(self, A, nnet_output, den_scores, tu_sum, mmi_weight): FILE: nets/scorers/mmi_rnnt_scorer.py class MMIRNNTScorer (line 11) | class MMIRNNTScorer(): method __init__ (line 12) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 36) | def load_weight(self, rank, path): method den_scores (line 47) | def den_scores(self, x): method batch_rescore (line 70) | def batch_rescore(self, A, h): method _batch_rescore (line 79) | def _batch_rescore(self, A, h): method batch_score (line 105) | def batch_score(self, A, nnet_output, den_scores, tu_sum, mmi_weight): FILE: nets/scorers/mmi_utils.py function intersect_dense_forward (line 26) | def intersect_dense_forward(a_fsas: Fsa, function step_trace (line 79) | def step_trace(out_fsas, a_fsas, arc_map_a, durations): function step_intersect (line 194) | def step_intersect(fsa, function step_intersect_test (line 223) | def step_intersect_test(): FILE: nets/scorers/new_mmi_frame_scorer.py class MMIFrameScorer (line 9) | class MMIFrameScorer(PartialScorerInterface): method __init__ (line 10) | def __init__(self, lang, device, idim, sos_id, rank, use_segment, char... method load_weight (line 31) | def load_weight(self, rank): method init_state (line 41) | def init_state(self, x): method select_state (line 64) | def select_state(self, states, j): method score (line 68) | def score(**kargs): method score_partial (line 71) | def score_partial(self, y, next_tokens, state, hs_pad): method final_score (line 122) | def final_score(self, state): FILE: nets/scorers/ngram.py class Ngrambase (line 12) | class Ngrambase(ABC): method __init__ (line 15) | def __init__(self, ngram_model, token_list): method init_state (line 28) | def init_state(self, x): method score_partial_ (line 34) | def score_partial_(self, y, next_token, state, x): class NgramFullScorer (line 60) | class NgramFullScorer(Ngrambase, BatchScorerInterface): method score (line 63) | def score(self, y, state, x): class NgramPartScorer (line 80) | class NgramPartScorer(Ngrambase, PartialScorerInterface): method score_partial (line 83) | def score_partial(self, y, next_token, state, x): method select_state (line 100) | def select_state(self, state, i): FILE: nets/scorers/sorted_matcher.py class SortedMatcher (line 4) | class SortedMatcher(object): method __init__ (line 15) | def __init__(self, vector_fst, max_num_arcs, method search (line 24) | def search(self, state_id, ilabel): method get_scores_wodisambig (line 58) | def get_scores_wodisambig(self, state_id, ilabel, init_score=0.0): method get_scores (line 82) | def get_scores(self, state_id, ilabel): method final_score (line 104) | def final_score(self, state_id): FILE: nets/scorers/tlg_scorer.py class TlgPartialScorer (line 13) | class TlgPartialScorer(PartialScorerInterface): method __init__ (line 19) | def __init__(self, lang, nonblk_reward=0.0): method init_state (line 39) | def init_state(self, x=None): method score_partial (line 45) | def score_partial(self, y, next_tokens, state, x): method score_one (line 70) | def score_one(self, tok_id, state_dict): method final_score (line 94) | def final_score(self, states): method final_score_one (line 109) | def final_score_one(self, state_dict): FILE: nets/scorers/trace_frame.py function trace_lattice (line 32) | def trace_lattice(lats): function compute_frame_level_scores (line 44) | def compute_frame_level_scores(graph, nnet_output): function trace_lattice_batch (line 73) | def trace_lattice_batch(lats, batch): function split_forward_scores (line 94) | def split_forward_scores(scores): function compute_frame_level_scores_batch (line 105) | def compute_frame_level_scores_batch(graphs, nnet_output): function compute_frame_level_scores_loop (line 134) | def compute_frame_level_scores_loop(graph, nnet_output): FILE: nets/scorers/word_ngram.py function is_disambig_symbol (line 18) | def is_disambig_symbol(symbol: str, pattern: re.Pattern = re.compile(r'^... function find_first_disambig_symbol (line 21) | def find_first_disambig_symbol(symbols: k2.SymbolTable) -> int: class WordNgram (line 41) | class WordNgram(): method __init__ (line 42) | def __init__(self, lang, device): method load_G (line 62) | def load_G(self): method text2lat (line 88) | def text2lat(self, text, gram_len=6): method score_lattice (line 129) | def score_lattice(self, lats, log_semiring=True): method score_texts (line 158) | def score_texts(self, texts, log_semiring=True): method draw (line 168) | def draw(self, fsavec, prefix=None): class WordNgramPartialScorer (line 175) | class WordNgramPartialScorer(PartialScorerInterface): method __init__ (line 176) | def __init__(self, lang, device, token_list, ignore_tokens=["", "... method init_state (line 201) | def init_state(self, x): method score_partial (line 204) | def score_partial(self, y, next_tokens, state, x): FILE: nets/st_interface.py class STInterface (line 7) | class STInterface(ASRInterface): method translate (line 15) | def translate(self, x, trans_args, char_list=None, rnnlm=None, ensembl... method translate_batch (line 27) | def translate_batch(self, x, trans_args, char_list=None, rnnlm=None): function dynamic_import_st (line 52) | def dynamic_import_st(module, backend): FILE: nets/transducer_decoder_interface.py class Hypothesis (line 15) | class Hypothesis: class NSCHypothesis (line 30) | class NSCHypothesis(Hypothesis): class TransducerDecoderInterface (line 37) | class TransducerDecoderInterface: method init_state (line 40) | def init_state( method score (line 59) | def score( method batch_score (line 82) | def batch_score( method select_state (line 109) | def select_state( method create_batch_states (line 130) | def create_batch_states( FILE: nets/tts_interface.py class Reporter (line 17) | class Reporter(chainer.Chain): method report (line 20) | def report(self, dicts): class TTSInterface (line 26) | class TTSInterface(object): method add_arguments (line 30) | def add_arguments(parser): method __init__ (line 34) | def __init__(self): method forward (line 38) | def forward(self, *args, **kwargs): method inference (line 47) | def inference(self, *args, **kwargs): method calculate_all_attentions (line 58) | def calculate_all_attentions(self, *args, **kwargs): method load_pretrained_model (line 67) | def load_pretrained_model(self, model_path): method attention_plot_class (line 72) | def attention_plot_class(self): method base_plot_keys (line 79) | def base_plot_keys(self): FILE: optimizer/chainer.py class AdamFactory (line 13) | class AdamFactory(OptimizerFactoryInterface): method add_arguments (line 17) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 22) | def from_args(target, args: argparse.Namespace): class SGDFactory (line 41) | class SGDFactory(OptimizerFactoryInterface): method add_arguments (line 45) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 50) | def from_args(target, args: argparse.Namespace): class AdadeltaFactory (line 67) | class AdadeltaFactory(OptimizerFactoryInterface): method add_arguments (line 71) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 76) | def from_args(target, args: argparse.Namespace): FILE: optimizer/factory.py class OptimizerFactoryInterface (line 8) | class OptimizerFactoryInterface: method from_args (line 12) | def from_args(target, args: argparse.Namespace): method add_arguments (line 24) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method build (line 29) | def build(cls, target, **kwargs): function dynamic_import_optimizer (line 45) | def dynamic_import_optimizer(name: str, backend: str) -> OptimizerFactor... FILE: optimizer/parser.py function sgd (line 4) | def sgd(parser): function adam (line 11) | def adam(parser): function adadelta (line 20) | def adadelta(parser): FILE: optimizer/pytorch.py class AdamFactory (line 12) | class AdamFactory(OptimizerFactoryInterface): method add_arguments (line 16) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 21) | def from_args(target, args: argparse.Namespace): class SGDFactory (line 38) | class SGDFactory(OptimizerFactoryInterface): method add_arguments (line 42) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 47) | def from_args(target, args: argparse.Namespace): class AdadeltaFactory (line 63) | class AdadeltaFactory(OptimizerFactoryInterface): method add_arguments (line 67) | def add_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen... method from_args (line 72) | def from_args(target, args: argparse.Namespace): FILE: scheduler/chainer.py class ChainerScheduler (line 10) | class ChainerScheduler: method __init__ (line 13) | def __init__(self, schedulers: List[SchedulerInterface], optimizer: Op... method step (line 21) | def step(self, n_iter: int): FILE: scheduler/pytorch.py class PyTorchScheduler (line 10) | class PyTorchScheduler: method __init__ (line 13) | def __init__(self, schedulers: List[SchedulerInterface], optimizer: Op... method step (line 21) | def step(self, n_iter: int): FILE: scheduler/scheduler.py class _PrefixParser (line 9) | class _PrefixParser: method __init__ (line 10) | def __init__(self, parser, prefix): method add_argument (line 14) | def add_argument(self, name, **kwargs): class SchedulerInterface (line 19) | class SchedulerInterface: method __init__ (line 24) | def __init__(self, key: str, args: argparse.Namespace): method get_arg (line 32) | def get_arg(self, name): method add_arguments (line 37) | def add_arguments(cls, key: str, parser: argparse.ArgumentParser): method _add_arguments (line 44) | def _add_arguments(parser: _PrefixParser): method build (line 48) | def build(cls, key: str, **kwargs): method scale (line 67) | def scale(self, n_iter: int) -> float: function register_scheduler (line 83) | def register_scheduler(cls): function dynamic_import_scheduler (line 89) | def dynamic_import_scheduler(module): class NoScheduler (line 107) | class NoScheduler(SchedulerInterface): method scale (line 112) | def scale(self, n_iter): class NoamScheduler (line 118) | class NoamScheduler(SchedulerInterface): method _add_arguments (line 129) | def _add_arguments(parser: _PrefixParser): method __init__ (line 135) | def __init__(self, key, args): method scale (line 140) | def scale(self, step): class CyclicCosineScheduler (line 147) | class CyclicCosineScheduler(SchedulerInterface): method _add_arguments (line 164) | def _add_arguments(parser: _PrefixParser): method scale (line 176) | def scale(self, n_iter): FILE: snowfall/common.py function setup_logger (line 26) | def setup_logger(log_filename: Pathlike, log_level: str = 'info', use_co... function load_checkpoint (line 57) | def load_checkpoint( function average_checkpoint (line 110) | def average_checkpoint(filenames: List[Pathlike], model: AcousticModel) ... function save_checkpoint (line 164) | def save_checkpoint( function save_training_info (line 200) | def save_training_info( function get_phone_symbols (line 228) | def get_phone_symbols(symbol_table: k2.SymbolTable, function cut_id_dumper (line 255) | def cut_id_dumper(dataloader, path: Path): function str2bool (line 274) | def str2bool(v): function describe (line 285) | def describe(model: torch.nn.Module): function get_texts (line 299) | def get_texts(best_paths: k2.Fsa, indices: Optional[torch.Tensor] = None... function invert_permutation (line 331) | def invert_permutation(indices: torch.Tensor) -> torch.Tensor: function find_first_disambig_symbol (line 337) | def find_first_disambig_symbol(symbols: k2.SymbolTable) -> int: function store_transcripts (line 341) | def store_transcripts(path: Pathlike, texts: Iterable[Tuple[str, str]]): function write_error_stats (line 347) | def write_error_stats(f: TextIO, test_set_name: str, results: List[Tuple... FILE: snowfall/data/aishell.py class AishellAsrDataModule (line 8) | class AishellAsrDataModule(AsrDataModule): method train_cuts (line 13) | def train_cuts(self) -> CutSet: method valid_cuts (line 18) | def valid_cuts(self) -> CutSet: method test_cuts (line 23) | def test_cuts(self) -> CutSet: FILE: snowfall/data/asr_datamodule.py class AsrDataModule (line 16) | class AsrDataModule(DataModule): method add_arguments (line 33) | def add_arguments(cls, parser: argparse.ArgumentParser): method train_dataloaders (line 90) | def train_dataloaders(self) -> DataLoader: method valid_dataloaders (line 162) | def valid_dataloaders(self) -> DataLoader: method test_dataloaders (line 200) | def test_dataloaders(self) -> Union[DataLoader, List[DataLoader]]: FILE: snowfall/data/datamodule.py class DataModule (line 9) | class DataModule: method __init__ (line 19) | def __init__(self, args: argparse.Namespace): method add_arguments (line 23) | def add_arguments(cls, parser: argparse.ArgumentParser): method train_cuts (line 26) | def train_cuts(self) -> Union[CutSet, List[CutSet]]: method valid_cuts (line 29) | def valid_cuts(self) -> Union[CutSet, List[CutSet]]: method test_cuts (line 32) | def test_cuts(self) -> Union[CutSet, List[CutSet]]: method train_dataloaders (line 35) | def train_dataloaders(self) -> Union[DataLoader, List[DataLoader]]: method valid_dataloaders (line 38) | def valid_dataloaders(self) -> Union[DataLoader, List[DataLoader]]: method test_dataloaders (line 41) | def test_dataloaders(self) -> Union[DataLoader, List[DataLoader]]: FILE: snowfall/data/librispeech.py class LibriSpeechAsrDataModule (line 13) | class LibriSpeechAsrDataModule(AsrDataModule): method add_arguments (line 20) | def add_arguments(cls, parser: argparse.ArgumentParser): method train_cuts (line 30) | def train_cuts(self) -> CutSet: method valid_cuts (line 42) | def valid_cuts(self) -> CutSet: method test_cuts (line 51) | def test_cuts(self) -> List[CutSet]: FILE: snowfall/decoding/graph.py function compile_HLG (line 8) | def compile_HLG( FILE: snowfall/decoding/lm_rescore.py function _intersect_device (line 11) | def _intersect_device(a_fsas: k2.Fsa, b_fsas: k2.Fsa, b_to_a_map: torch.... function compute_am_scores (line 51) | def compute_am_scores(lats: k2.Fsa, word_fsas_with_epsilon_loops: k2.Fsa, function rescore_with_n_best_list (line 106) | def rescore_with_n_best_list(lats: k2.Fsa, G: k2.Fsa, function rescore_with_whole_lattice (line 236) | def rescore_with_whole_lattice(lats: k2.Fsa, function decode_with_lm_rescoring (line 300) | def decode_with_lm_rescoring(lats: k2.Fsa, G: k2.Fsa, num_paths: int, FILE: snowfall/dist.py function setup_dist (line 6) | def setup_dist(rank, world_size, master_port = None): function cleanup_dist (line 14) | def cleanup_dist(): FILE: snowfall/lexicon.py class Lexicon (line 13) | class Lexicon: method __init__ (line 14) | def __init__(self, lang_dir: Path): method phone_symbols (line 29) | def phone_symbols( FILE: snowfall/models/conformer.py class Conformer (line 18) | class Conformer(Transformer): method __init__ (line 35) | def __init__(self, num_features: int, num_classes: int, subsampling_fa... method encode (line 50) | def encode(self, x: Tensor, supervisions: Optional[Dict] = None) -> Tu... class ConformerEncoderLayer (line 72) | class ConformerEncoderLayer(nn.Module): method __init__ (line 92) | def __init__(self, d_model: int, nhead: int, dim_feedforward: int = 20... method forward (line 126) | def forward(self, src: Tensor, pos_emb: Tensor, src_mask: Optional[Ten... class ConformerEncoder (line 185) | class ConformerEncoder(nn.TransformerEncoder): method __init__ (line 201) | def __init__(self, encoder_layer: nn.Module, num_layers: int, norm: nn... method forward (line 204) | def forward(self, src: Tensor, pos_emb: Tensor, mask: Optional[Tensor]... class RelPositionalEncoding (line 232) | class RelPositionalEncoding(torch.nn.Module): method __init__ (line 245) | def __init__(self, d_model: int, dropout_rate: float, max_len: int = 5... method extend_pe (line 254) | def extend_pe(self, x: Tensor) -> None: method forward (line 286) | def forward(self, x: torch.Tensor) -> Tuple[Tensor, Tensor]: class RelPositionMultiheadAttention (line 306) | class RelPositionMultiheadAttention(nn.Module): method __init__ (line 322) | def __init__(self, embed_dim: int, num_heads: int, dropout: float = 0.... method _reset_parameters (line 342) | def _reset_parameters(self) -> None: method forward (line 350) | def forward(self, query: Tensor, key: Tensor, value: Tensor, pos_emb: ... method rel_shift (line 401) | def rel_shift(self, x: Tensor) -> Tensor: method multi_head_attention_forward (line 420) | def multi_head_attention_forward(self, query: Tensor, class ConvolutionModule (line 636) | class ConvolutionModule(nn.Module): method __init__ (line 647) | def __init__(self, channels: int, kernel_size: int, bias: bool = True)... method forward (line 681) | def forward(self, x: Tensor) -> Tensor: class Swish (line 707) | class Swish(torch.nn.Module): method forward (line 710) | def forward(self, x: Tensor) -> Tensor: FILE: snowfall/models/contextnet.py class ContextNet (line 14) | class ContextNet(AcousticModel): method __init__ (line 31) | def __init__( method forward (line 76) | def forward(self, x, supervision = None): class ContextNetBlock (line 93) | class ContextNetBlock(torch.nn.Module): method __init__ (line 106) | def __init__( method forward (line 145) | def forward(self, x): class SEModule (line 162) | class SEModule(torch.nn.Module): method __init__ (line 171) | def __init__( method forward (line 192) | def forward(self, x): class ConvModule (line 212) | class ConvModule(torch.nn.Module): method __init__ (line 226) | def __init__( method forward (line 258) | def forward(self, x): class SeparableConv1D (line 274) | class SeparableConv1D(nn.Module): method __init__ (line 289) | def __init__( method forward (line 320) | def forward(self, x): FILE: snowfall/models/interface.py class AcousticModel (line 7) | class AcousticModel(nn.Module): method write_tensorboard_diagnostics (line 25) | def write_tensorboard_diagnostics( FILE: snowfall/models/tdnn.py class Tdnn1a (line 14) | class Tdnn1a(AcousticModel): method __init__ (line 21) | def __init__(self, num_features: int, num_classes: int, subsampling_fa... method forward (line 93) | def forward(self, x: Tensor) -> Tensor: method write_tensorboard_diagnostics (line 106) | def write_tensorboard_diagnostics( FILE: snowfall/models/tdnn_lstm.py class TdnnLstm1a (line 11) | class TdnnLstm1a(AcousticModel): method __init__ (line 18) | def __init__(self, num_features: int, num_classes: int, subsampling_fa... method forward (line 89) | def forward(self, x: Tensor) -> Tensor: class TdnnLstm1b (line 106) | class TdnnLstm1b(AcousticModel): method __init__ (line 113) | def __init__(self, num_features: int, num_classes: int, subsampling_fa... method forward (line 149) | def forward(self, x: Tensor) -> Tensor: method write_tensorboard_diagnostics (line 170) | def write_tensorboard_diagnostics( FILE: snowfall/models/tdnnf.py function tdnnf_optimizer (line 23) | def tdnnf_optimizer( class Tdnnf1a (line 46) | class Tdnnf1a(AcousticModel): method __init__ (line 76) | def __init__(self, method forward (line 133) | def forward(self, x, dropout=0.): method write_tensorboard_diagnostics (line 161) | def write_tensorboard_diagnostics(self, tb_writer: SummaryWriter, glob... function constrain_orthonormal_hook (line 179) | def constrain_orthonormal_hook(model, unused_x): function _constrain_orthonormal_internal (line 193) | def _constrain_orthonormal_internal(M): class SharedDimScaleDropout (line 235) | class SharedDimScaleDropout(nn.Module): method __init__ (line 236) | def __init__(self, dim=1): method forward (line 245) | def forward(self, x, alpha=0.0): class OrthonormalLinear (line 257) | class OrthonormalLinear(nn.Module): method __init__ (line 259) | def __init__(self, dim, bottleneck_dim, kernel_size): method forward (line 273) | def forward(self, x): method constrain_orthonormal (line 279) | def constrain_orthonormal(self): class PrefinalLayer (line 308) | class PrefinalLayer(nn.Module): method __init__ (line 310) | def __init__(self, big_dim, small_dim): method forward (line 319) | def forward(self, x): class TDNN (line 343) | class TDNN(nn.Module): method __init__ (line 349) | def __init__(self, input_dim, hidden_dim): method forward (line 362) | def forward(self, x, dropout=0.): class FactorizedTDNN (line 372) | class FactorizedTDNN(nn.Module): method __init__ (line 382) | def __init__(self, method forward (line 416) | def forward(self, x, dropout=0.): FILE: snowfall/models/transformer.py class Transformer (line 16) | class Transformer(AcousticModel): method __init__ (line 32) | def __init__(self, num_features: int, num_classes: int, subsampling_fa... method forward (line 83) | def forward(self, x: Tensor, supervision: Optional[Dict] = None) -> Tu... method encode (line 99) | def encode(self, x: Tensor, supervisions: Optional[Dict] = None) -> Tu... method encoder_output (line 120) | def encoder_output(self, x: Tensor) -> Tensor: method decoder_forward (line 132) | def decoder_forward(self, x: Tensor, encoder_mask: Tensor, supervision... class TransformerEncoderLayer (line 170) | class TransformerEncoderLayer(nn.Module): method __init__ (line 189) | def __init__(self, d_model: int, nhead: int, dim_feedforward: int = 20... method __setstate__ (line 207) | def __setstate__(self, state): method forward (line 212) | def forward(self, src: Tensor, src_mask: Optional[Tensor] = None, class TransformerDecoderLayer (line 247) | class TransformerDecoderLayer(nn.Module): method __init__ (line 266) | def __init__(self, d_model: int, nhead: int, dim_feedforward: int = 20... method __setstate__ (line 287) | def __setstate__(self, state): method forward (line 292) | def forward(self, tgt: Tensor, memory: Tensor, tgt_mask: Optional[Tens... function _get_activation_fn (line 343) | def _get_activation_fn(activation: str): class Conv2dSubsampling (line 352) | class Conv2dSubsampling(nn.Module): method __init__ (line 362) | def __init__(self, idim: int, odim: int) -> None: method forward (line 373) | def forward(self, x: Tensor) -> Tensor: class VggSubsampling (line 391) | class VggSubsampling(nn.Module): method __init__ (line 402) | def __init__(self, idim: int, odim: int) -> None: method forward (line 440) | def forward(self, x: Tensor) -> Tensor: class PositionalEncoding (line 458) | class PositionalEncoding(nn.Module): method __init__ (line 469) | def __init__(self, d_model: int, dropout: float = 0.1, max_len: int = ... method extend_pe (line 478) | def extend_pe(self, x: Tensor) -> None: method forward (line 496) | def forward(self, x: Tensor) -> Tensor: class Noam (line 512) | class Noam(object): method __init__ (line 524) | def __init__(self, params, model_size: int = 256, factor: float = 10.0... method param_groups (line 534) | def param_groups(self): method step (line 538) | def step(self): method rate (line 547) | def rate(self, step=None): method zero_grad (line 557) | def zero_grad(self): method state_dict (line 561) | def state_dict(self): method load_state_dict (line 572) | def load_state_dict(self, state_dict): class LabelSmoothingLoss (line 581) | class LabelSmoothingLoss(nn.Module): method __init__ (line 595) | def __init__( method forward (line 614) | def forward(self, x: Tensor, target: Tensor) -> Tensor: function encoder_padding_mask (line 641) | def encoder_padding_mask(max_len: int, supervisions: Optional[Dict] = No... function decoder_padding_mask (line 673) | def decoder_padding_mask(ys_pad: Tensor, ignore_id: int = -1) -> Tensor: function get_normal_transcripts (line 688) | def get_normal_transcripts(supervision: Dict, words: k2.SymbolTable, oov... function generate_square_subsequent_mask (line 711) | def generate_square_subsequent_mask(sz: int) -> Tensor: function add_sos_eos (line 726) | def add_sos_eos(ys: List[List[int]], lexicon: k2.Fsa, sos: int, eos: int... function pad_list (line 749) | def pad_list(ys: List[Tensor], pad_value: float) -> Tensor: function get_hierarchical_targets (line 779) | def get_hierarchical_targets(ys: List[List[int]], lexicon: k2.Fsa) -> Li... function test_transformer (line 820) | def test_transformer(): function main (line 827) | def main(): FILE: snowfall/objectives/common.py function encode_supervisions (line 7) | def encode_supervisions(supervisions: Dict[str, Tensor]) -> Tuple[Tensor... function get_tot_objf_and_num_frames (line 35) | def get_tot_objf_and_num_frames( FILE: snowfall/objectives/ctc.py class CTCLoss (line 12) | class CTCLoss(nn.Module): method __init__ (line 18) | def __init__( method forward (line 25) | def forward( FILE: snowfall/objectives/mmi.py function _compute_mmi_loss_exact_optimized (line 12) | def _compute_mmi_loss_exact_optimized( function _compute_mmi_loss_exact_non_optimized (line 110) | def _compute_mmi_loss_exact_non_optimized( function _compute_mmi_loss_pruned (line 146) | def _compute_mmi_loss_pruned( class LFMMILoss (line 196) | class LFMMILoss(nn.Module): method __init__ (line 203) | def __init__( method forward (line 216) | def forward(self, nnet_output: torch.Tensor, texts: List[str], FILE: snowfall/training/ctc_graph.py function build_ctc_topo (line 13) | def build_ctc_topo(tokens: List[int]) -> k2.Fsa: class CtcTrainingGraphCompiler (line 45) | class CtcTrainingGraphCompiler(object): method __init__ (line 47) | def __init__(self, method compile (line 76) | def compile(self, texts: Iterable[str]) -> k2.Fsa: method compile_one_and_cache (line 85) | def compile_one_and_cache(self, text: str) -> k2.Fsa: FILE: snowfall/training/diagnostics.py function l1_norm (line 8) | def l1_norm(x): function l2_norm (line 12) | def l2_norm(x): function linf_norm (line 16) | def linf_norm(x): function measure_weight_norms (line 20) | def measure_weight_norms(model: nn.Module, norm: str = 'l2') -> Dict[str... function measure_semiorthogonality (line 43) | def measure_semiorthogonality(model: nn.Module) -> Dict[str, float]: function measure_gradient_norms (line 67) | def measure_gradient_norms(model: nn.Module, norm: str = 'l1') -> Dict[s... function optim_step_and_measure_param_change (line 90) | def optim_step_and_measure_param_change( FILE: snowfall/training/mmi_graph.py function create_bigram_phone_lm (line 15) | def create_bigram_phone_lm(phones: List[int]) -> k2.Fsa: class MmiTrainingGraphCompiler (line 44) | class MmiTrainingGraphCompiler(object): method __init__ (line 46) | def __init__( method compile (line 86) | def compile(self, method build_num_graphs (line 146) | def build_num_graphs(self, texts: List[str]) -> k2.Fsa: method compile_lookahead_numerators (line 179) | def compile_lookahead_numerators(self, word_fsa_vec, P): FILE: snowfall/training/mmi_mbr_graph.py function find_first_disambig_symbol (line 19) | def find_first_disambig_symbol(symbols: k2.SymbolTable) -> int: class MmiMbrTrainingGraphCompiler (line 23) | class MmiMbrTrainingGraphCompiler(object): method __init__ (line 25) | def __init__(self, method compile (line 109) | def compile(self, texts: Iterable[str], method build_num_graphs (line 163) | def build_num_graphs(self, texts: List[str]) -> k2.Fsa: FILE: snowfall/warpper/k2_decode.py function k2_decode (line 12) | def k2_decode(model, device, js, sampler, batch_size, use_segment=False): function build_batch_data (line 48) | def build_batch_data(feats): function parse_results (line 66) | def parse_results(results): FILE: snowfall/warpper/mmi_test.py function main (line 8) | def main(): FILE: snowfall/warpper/mmi_utils.py function build_word_mapping (line 3) | def build_word_mapping(word_mapping): function convert_transcription (line 10) | def convert_transcription(ys, mapping, words, oov_id, ignore_ids): function encode_supervision (line 30) | def encode_supervision(hlens): function parse_step (line 40) | def parse_step(hyp, words, part_ids, weights, full_scores, part_scores, ... FILE: snowfall/warpper/prefix_scorer.py function build_word_fsa (line 11) | def build_word_fsa(prefix_ids, candidate_intervals): FILE: snowfall/warpper/warpper_ctc.py class K2CTC (line 29) | class K2CTC(torch.nn.Module): method __init__ (line 31) | def __init__(self, method log_softmax (line 95) | def log_softmax(self, hs_pad): method softmax (line 98) | def softmax(self, hs_pad): method argmax (line 103) | def argmax(self, hs_pad): method forward (line 106) | def forward(self, hs_pad, hlens, ys_pad, texts): method decode (line 134) | def decode(self, nnet_output, hlens, texts, is_english): method decode_init (line 206) | def decode_init(self): method dump_weight (line 240) | def dump_weight(self, rank): FILE: snowfall/warpper/warpper_mmi.py class K2MMI (line 30) | class K2MMI(torch.nn.Module): method __init__ (line 32) | def __init__(self, method log_softmax (line 104) | def log_softmax(self, hs_pad): method softmax (line 107) | def softmax(self, hs_pad): method argmax (line 112) | def argmax(self, hs_pad): method forward (line 115) | def forward(self, hs_pad, hlens, ys_pad, texts): method decode (line 153) | def decode(self, nnet_output, hlens, texts, is_english): method decode_init (line 225) | def decode_init(self): method dump_weight (line 259) | def dump_weight(self, rank, path): FILE: st/pytorch_backend/st.py class CustomConverter (line 56) | class CustomConverter(ASRCustomConverter): method __init__ (line 66) | def __init__( method __call__ (line 73) | def __call__(self, batch, device=torch.device("cpu")): function train (line 112) | def train(args): function trans (line 604) | def trans(args): FILE: transform/add_deltas.py function delta (line 4) | def delta(feat, window): function add_deltas (line 16) | def add_deltas(x, window=2, order=2): class AddDeltas (line 23) | class AddDeltas(object): method __init__ (line 24) | def __init__(self, window=2, order=2): method __repr__ (line 28) | def __repr__(self): method __call__ (line 33) | def __call__(self, x): FILE: transform/channel_selector.py class ChannelSelector (line 4) | class ChannelSelector(object): method __init__ (line 7) | def __init__(self, train_channel="random", eval_channel=0, axis=1): method __repr__ (line 12) | def __repr__(self): method __call__ (line 23) | def __call__(self, x, train=True): FILE: transform/cmvn.py class CMVN (line 8) | class CMVN(object): method __init__ (line 9) | def __init__( method __repr__ (line 84) | def __repr__(self): method __call__ (line 97) | def __call__(self, x, uttid=None): class UtteranceCMVN (line 118) | class UtteranceCMVN(object): method __init__ (line 119) | def __init__(self, norm_means=True, norm_vars=False, std_floor=1.0e-20): method __repr__ (line 124) | def __repr__(self): method __call__ (line 131) | def __call__(self, x, uttid=None): FILE: transform/functional.py class FuncTrans (line 7) | class FuncTrans(TransformInterface): method __init__ (line 32) | def __init__(self, **kwargs): method __call__ (line 36) | def __call__(self, x): method add_arguments (line 40) | def add_arguments(cls, parser): method func (line 50) | def func(self): method default_params (line 54) | def default_params(cls): method __repr__ (line 63) | def __repr__(self): FILE: transform/perturb.py class SpeedPerturbation (line 9) | class SpeedPerturbation(object): method __init__ (line 25) | def __init__( method __repr__ (line 57) | def __repr__(self): method __call__ (line 71) | def __call__(self, x, uttid=None, train=True): class BandpassPerturbation (line 101) | class BandpassPerturbation(object): method __init__ (line 115) | def __init__(self, lower=0.0, upper=0.75, seed=None, axes=(-1,)): method __repr__ (line 122) | def __repr__(self): method __call__ (line 127) | def __call__(self, x_stft, uttid=None, train=True): class VolumePerturbation (line 145) | class VolumePerturbation(object): method __init__ (line 146) | def __init__(self, lower=-1.6, upper=1.6, utt2ratio=None, dbunit=True,... method __repr__ (line 169) | def __repr__(self): method __call__ (line 179) | def __call__(self, x, uttid=None, train=True): class NoiseInjection (line 194) | class NoiseInjection(object): method __init__ (line 197) | def __init__( method __repr__ (line 250) | def __repr__(self): method __call__ (line 260) | def __call__(self, x, uttid=None, train=True): class RIRConvolve (line 303) | class RIRConvolve(object): method __init__ (line 304) | def __init__(self, utt2rir, filetype="list"): method __repr__ (line 321) | def __repr__(self): method __call__ (line 324) | def __call__(self, x, uttid=None, train=True): FILE: transform/spec_augment.py function time_warp (line 12) | def time_warp(x, max_time_warp=80, inplace=False, mode="PIL"): class TimeWarp (line 54) | class TimeWarp(FuncTrans): method __call__ (line 58) | def __call__(self, x, train): function freq_mask (line 64) | def freq_mask(x, F=30, n_mask=2, replace_with_zero=True, inplace=False): class FreqMask (line 95) | class FreqMask(FuncTrans): method __call__ (line 99) | def __call__(self, x, train): function time_mask (line 105) | def time_mask(spec, T=40, n_mask=2, replace_with_zero=True, inplace=False): class TimeMask (line 137) | class TimeMask(FuncTrans): method __call__ (line 141) | def __call__(self, x, train): function spec_augment (line 147) | def spec_augment( class SpecAugment (line 195) | class SpecAugment(FuncTrans): method __call__ (line 199) | def __call__(self, x, train): FILE: transform/spectrogram.py function stft (line 5) | def stft( function istft (line 41) | def istft(x, n_shift, win_length=None, window="hann", center=True): function stft2logmelspectrogram (line 71) | def stft2logmelspectrogram(x_stft, fs, n_mels, n_fft, fmin=None, fmax=No... function spectrogram (line 86) | def spectrogram(x, n_fft, n_shift, win_length=None, window="hann"): function logmelspectrogram (line 92) | def logmelspectrogram( class Spectrogram (line 120) | class Spectrogram(object): method __init__ (line 121) | def __init__(self, n_fft, n_shift, win_length=None, window="hann"): method __repr__ (line 127) | def __repr__(self): method __call__ (line 139) | def __call__(self, x): class LogMelSpectrogram (line 149) | class LogMelSpectrogram(object): method __init__ (line 150) | def __init__( method __repr__ (line 172) | def __repr__(self): method __call__ (line 190) | def __call__(self, x): class Stft2LogMelSpectrogram (line 202) | class Stft2LogMelSpectrogram(object): method __init__ (line 203) | def __init__(self, fs, n_mels, n_fft, fmin=None, fmax=None, eps=1e-10): method __repr__ (line 211) | def __repr__(self): method __call__ (line 225) | def __call__(self, x): class Stft (line 236) | class Stft(object): method __init__ (line 237) | def __init__( method __repr__ (line 253) | def __repr__(self): method __call__ (line 268) | def __call__(self, x): class IStft (line 280) | class IStft(object): method __init__ (line 281) | def __init__(self, n_shift, win_length=None, window="hann", center=True): method __repr__ (line 287) | def __repr__(self): method __call__ (line 300) | def __call__(self, x): FILE: transform/transform_interface.py class TransformInterface (line 2) | class TransformInterface: method __call__ (line 5) | def __call__(self, x): method add_arguments (line 9) | def add_arguments(cls, parser): method __repr__ (line 12) | def __repr__(self): class Identity (line 16) | class Identity(TransformInterface): method __call__ (line 19) | def __call__(self, x): FILE: transform/transformation.py class Transformation (line 49) | class Transformation(object): method __init__ (line 67) | def __init__(self, conffile=None): method __repr__ (line 106) | def __repr__(self): method __call__ (line 112) | def __call__(self, xs, uttid_list=None, **kwargs): FILE: transform/wpe.py class WPE (line 4) | class WPE(object): method __init__ (line 5) | def __init__( method __repr__ (line 14) | def __repr__(self): method __call__ (line 28) | def __call__(self, xs): FILE: tts/pytorch_backend/tts.py class CustomEvaluator (line 53) | class CustomEvaluator(BaseEvaluator): method __init__ (line 56) | def __init__(self, model, iterator, target, device): method evaluate (line 71) | def evaluate(self): class CustomUpdater (line 108) | class CustomUpdater(training.StandardUpdater): method __init__ (line 111) | def __init__(self, model, grad_clip, iterator, optimizer, device, accu... method update_core (line 131) | def update_core(self): method update (line 169) | def update(self): class CustomConverter (line 176) | class CustomConverter(object): method __init__ (line 179) | def __init__(self): method __call__ (line 184) | def __call__(self, batch, device=torch.device("cpu")): function train (line 263) | def train(args): function decode (line 570) | def decode(args): FILE: utils/bmuf.py function _copy_vec_to_param (line 14) | def _copy_vec_to_param(vec, parameters, is_grad=False): class BmufTrainer (line 42) | class BmufTrainer(): method __init__ (line 55) | def __init__(self, master_node, rank, world_size, model, method update_and_sync (line 80) | def update_and_sync(self): method broadcast (line 109) | def broadcast(self, tensor): method sum_reduce (line 113) | def sum_reduce(self, tensor): class BlockAdamTrainer (line 119) | class BlockAdamTrainer(): method __init__ (line 139) | def __init__(self, args, master_node, rank, world_size, model): method update_and_sync (line 164) | def update_and_sync(self): class BmufAdamTrainer (line 255) | class BmufAdamTrainer(): method __init__ (line 270) | def __init__(self, master_node, rank, world_size, model, method update_and_sync (line 315) | def update_and_sync(self): method broadcast (line 390) | def broadcast(self, tensor): method sum_reduce (line 394) | def sum_reduce(self, tensor): FILE: utils/check_kwargs.py function check_kwargs (line 4) | def check_kwargs(func, kwargs, name=None): FILE: utils/cli_readers.py function file_reader_helper (line 12) | def file_reader_helper( class KaldiReader (line 55) | class KaldiReader: method __init__ (line 56) | def __init__(self, rspecifier, return_shape=False, segments=None): method __iter__ (line 61) | def __iter__(self): class HDF5Reader (line 69) | class HDF5Reader: method __init__ (line 70) | def __init__(self, rspecifier, return_shape=False): method __iter__ (line 82) | def __iter__(self): class SoundHDF5Reader (line 141) | class SoundHDF5Reader: method __init__ (line 142) | def __init__(self, rspecifier, return_shape=False): method __iter__ (line 152) | def __iter__(self): class SoundReader (line 211) | class SoundReader: method __init__ (line 212) | def __init__(self, rspecifier, return_shape=False): method __iter__ (line 224) | def __iter__(self): FILE: utils/cli_utils.py function strtobool (line 8) | def strtobool(x): function get_commandline_args (line 13) | def get_commandline_args(): function is_scipy_wav_style (line 48) | def is_scipy_wav_style(value): function assert_scipy_wav_style (line 58) | def assert_scipy_wav_style(value): FILE: utils/cli_writers.py function file_writer_helper (line 13) | def file_writer_helper( class BaseWriter (line 82) | class BaseWriter: method __setitem__ (line 83) | def __setitem__(self, key, value): method __enter__ (line 86) | def __enter__(self): method __exit__ (line 89) | def __exit__(self, exc_type, exc_val, exc_tb): method close (line 92) | def close(self): function get_num_frames_writer (line 111) | def get_num_frames_writer(write_num_frames: str): class KaldiWriter (line 134) | class KaldiWriter(BaseWriter): method __init__ (line 135) | def __init__( method __setitem__ (line 150) | def __setitem__(self, key, value): function parse_wspecifier (line 156) | def parse_wspecifier(wspecifier: str) -> Dict[str, str]: class HDF5Writer (line 175) | class HDF5Writer(BaseWriter): method __init__ (line 183) | def __init__(self, wspecifier, write_num_frames=None, compress=False): method __setitem__ (line 201) | def __setitem__(self, key, value): class SoundHDF5Writer (line 210) | class SoundHDF5Writer(BaseWriter): method __init__ (line 219) | def __init__(self, wspecifier, write_num_frames=None, pcm_format="wav"): method __setitem__ (line 233) | def __setitem__(self, key, value): class SoundWriter (line 246) | class SoundWriter(BaseWriter): method __init__ (line 255) | def __init__(self, wspecifier, write_num_frames=None, pcm_format="wav"): method __setitem__ (line 273) | def __setitem__(self, key, value): FILE: utils/dataset.py function get_time (line 12) | def get_time(): class TransformDataset (line 15) | class TransformDataset(torch.utils.data.Dataset): method __init__ (line 24) | def __init__(self, data, transform): method __len__ (line 30) | def __len__(self): method __getitem__ (line 34) | def __getitem__(self, idx): class ChainerDataLoader (line 40) | class ChainerDataLoader(object): method __init__ (line 48) | def __init__(self, **kwargs): method next (line 57) | def next(self): method __iter__ (line 72) | def __iter__(self): method epoch_detail (line 78) | def epoch_detail(self): method serialize (line 82) | def serialize(self, serializer): method start_shuffle (line 89) | def start_shuffle(self): method finalize (line 94) | def finalize(self): FILE: utils/deterministic_utils.py function set_deterministic_pytorch (line 8) | def set_deterministic_pytorch(args): function set_deterministic_chainer (line 34) | def set_deterministic_chainer(args): FILE: utils/draw_num_fst.py function main (line 11) | def main(): FILE: utils/dynamic_import.py function dynamic_import (line 4) | def dynamic_import(import_path, alias=dict()): FILE: utils/fill_missing_args.py function fill_missing_args (line 10) | def fill_missing_args(args, add_arguments): FILE: utils/io_utils.py function get_time (line 15) | def get_time(): class LoadInputsAndTargets (line 18) | class LoadInputsAndTargets(object): method __init__ (line 44) | def __init__( method __call__ (line 99) | def __call__(self, batch, return_uttid=False): method _create_batch_asr (line 227) | def _create_batch_asr(self, x_feats_dict, y_feats_dict, uttid_list): method _create_batch_mt (line 289) | def _create_batch_mt(self, x_feats_dict, y_feats_dict, uttid_list): method _create_batch_tts (line 336) | def _create_batch_tts(self, x_feats_dict, y_feats_dict, uttid_list, eos): method _create_batch_vc (line 413) | def _create_batch_vc(self, x_feats_dict, y_feats_dict, uttid_list): method _get_from_loader (line 491) | def _get_from_loader(self, filepath, filetype, uttid=None): class SoundHDF5File (line 618) | class SoundHDF5File(object): method __init__ (line 634) | def __init__(self, filepath, mode="r+", format=None, dtype="int16", **... method __repr__ (line 651) | def __repr__(self): method create_dataset (line 656) | def create_dataset(self, name, shape=None, data=None, **kwds): method __setitem__ (line 662) | def __setitem__(self, name, data): method __getitem__ (line 665) | def __getitem__(self, key): method keys (line 671) | def keys(self): method values (line 674) | def values(self): method items (line 678) | def items(self): method __iter__ (line 682) | def __iter__(self): method __contains__ (line 685) | def __contains__(self, item): method __len__ (line 688) | def __len__(self, item): method __enter__ (line 691) | def __enter__(self): method __exit__ (line 694) | def __exit__(self, exc_type, exc_val, exc_tb): method close (line 697) | def close(self): function parse_arkpath (line 701) | def parse_arkpath(path): function load_ark_full (line 705) | def load_ark_full(path): function wid_suffix (line 712) | def wid_suffix(): function memory_ratio (line 716) | def memory_ratio(): FILE: utils/parse_decoding_process.py function plot_decoding_logs (line 6) | def plot_decoding_logs(graph_dir, char_list, recog_args, uttid, nbest_hy... function plot_dict (line 26) | def plot_dict(filename, d, title): function process_logs (line 39) | def process_logs(logs, args, accum=False): FILE: utils/parse_npy.py function main (line 4) | def main(): FILE: utils/print.py function step_print (line 5) | def step_print(ctx, flush=False): FILE: utils/rtf_calculator.py class RTF_calculator (line 3) | class RTF_calculator(): method __init__ (line 4) | def __init__(self, js, fps=100): method tik (line 9) | def tik(self): method tok (line 12) | def tok(self): FILE: utils/sampler.py class BufferSampler (line 8) | class BufferSampler(object): method __init__ (line 9) | def __init__(self, length, utts_per_ark, batch_size, buf_size, seed=0,... method __iter__ (line 37) | def __iter__(self): method __len__ (line 44) | def __len__(self): method _get_indices (line 65) | def _get_indices(self): method _get_indices_deprecated (line 124) | def _get_indices_deprecated(self): method reset (line 152) | def reset(self, seed=None): method _splice_list (line 164) | def _splice_list(self, lsts): method get_prefetch_factor (line 176) | def get_prefetch_factor(self): class testdataset (line 179) | class testdataset: method __init__ (line 180) | def __init__(self, length): method __len__ (line 183) | def __len__(self): FILE: utils/spec_augment.py function specaug (line 34) | def specaug( function time_warp (line 67) | def time_warp(spec, W=5): function freq_mask (line 95) | def freq_mask(spec, F=30, num_masks=1, replace_with_zero=False): function time_mask (line 123) | def time_mask(spec, T=40, num_masks=1, replace_with_zero=False): function sparse_image_warp (line 151) | def sparse_image_warp( function get_grid_locations (line 184) | def get_grid_locations(image_height, image_width, device): function flatten_grid_locations (line 191) | def flatten_grid_locations(grid_locations, image_height, image_width): function get_flat_grid_locations (line 195) | def get_flat_grid_locations(image_height, image_width, device): function create_dense_flows (line 202) | def create_dense_flows(flattened_flows, batch_size, image_height, image_... function interpolate_spline (line 207) | def interpolate_spline( function solve_interpolation (line 222) | def solve_interpolation(train_points, train_values, order, regularizatio... function cross_squared_distance_matrix (line 262) | def cross_squared_distance_matrix(x, y): function phi (line 284) | def phi(r, order): function apply_interpolation (line 313) | def apply_interpolation(query_points, train_points, w, v, order): function dense_image_warp (line 349) | def dense_image_warp(image, flow): function interpolate_bilinear (line 400) | def interpolate_bilinear( FILE: utils/training/batchfy.py function batchfy_by_seq (line 6) | def batchfy_by_seq( function batchfy_by_bin (line 93) | def batchfy_by_bin( function batchfy_by_frame (line 184) | def batchfy_by_frame( function batchfy_shuffle (line 293) | def batchfy_shuffle(data, batch_size, min_batch_size, num_batches, short... function make_batchset (line 332) | def make_batchset( FILE: utils/training/evaluator.py class BaseEvaluator (line 6) | class BaseEvaluator(Evaluator): method __call__ (line 9) | def __call__(self, trainer=None): FILE: utils/training/iterators.py class ShufflingEnabler (line 10) | class ShufflingEnabler(Extension): method __init__ (line 13) | def __init__(self, iterators): method __call__ (line 21) | def __call__(self, trainer): class ToggleableShufflingSerialIterator (line 32) | class ToggleableShufflingSerialIterator(SerialIterator): method __init__ (line 35) | def __init__(self, dataset, batch_size, repeat=True, shuffle=True): method start_shuffle (line 47) | def start_shuffle(self): class ToggleableShufflingMultiprocessIterator (line 57) | class ToggleableShufflingMultiprocessIterator(MultiprocessIterator): method __init__ (line 60) | def __init__( method start_shuffle (line 93) | def start_shuffle(self): FILE: utils/training/tensorboard_logger.py class TensorboardLogger (line 4) | class TensorboardLogger(Extension): method __init__ (line 9) | def __init__( method __call__ (line 25) | def __call__(self, trainer): FILE: utils/training/train_utils.py function check_early_stop (line 5) | def check_early_stop(trainer, epochs): function set_early_stop (line 20) | def set_early_stop(trainer, args, is_lm=False): FILE: vc/pytorch_backend/vc.py class CustomEvaluator (line 53) | class CustomEvaluator(BaseEvaluator): method __init__ (line 56) | def __init__(self, model, iterator, target, device): method evaluate (line 71) | def evaluate(self): class CustomUpdater (line 108) | class CustomUpdater(training.StandardUpdater): method __init__ (line 111) | def __init__(self, model, grad_clip, iterator, optimizer, device, accu... method update_core (line 131) | def update_core(self): method update (line 169) | def update(self): class CustomConverter (line 176) | class CustomConverter(object): method __init__ (line 179) | def __init__(self): method __call__ (line 184) | def __call__(self, batch, device=torch.device("cpu")): function train (line 263) | def train(args): function decode (line 564) | def decode(args):