Skip to content

Commit

Permalink
Fix #3
Browse files Browse the repository at this point in the history
  • Loading branch information
galv committed Oct 20, 2022
1 parent 77f0876 commit 304b54a
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .gitmodules
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[submodule "third_party/kaldi"]
path = third_party/kaldi
url = https://gitlab-master.nvidia.com/belgarten/kaldi.git
url = https://github.com/galv/kaldi.git
[submodule "third_party/pybind11"]
path = third_party/pybind11
url = https://github.com/pybind/pybind11.git
Expand Down
4 changes: 0 additions & 4 deletions src/riva/asrlib/decoder/test_graph_construction.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,17 +157,13 @@ def run_decoder(self, graph_path: str):

config = BatchedMappedDecoderCudaConfig()
config.n_input_per_chunk = 50
config.online_opts.lattice_postprocessor_opts.word_boundary_rxfilename = str(
os.path.join(graph_path, "graph/graph_ctc_3-gram.pruned.3e-7/phones/word_boundary.int")
)
config.online_opts.decoder_opts.default_beam = 17.0
config.online_opts.decoder_opts.lattice_beam = 8.0
config.online_opts.decoder_opts.max_active = 7000
config.online_opts.determinize_lattice = True
config.online_opts.max_batch_size = 400
config.online_opts.num_channels = 800
config.online_opts.frame_shift_seconds = 0.03
config.online_opts.lattice_postprocessor_opts.max_expand = 10
decoder = BatchedMappedDecoderCuda(
config,
os.path.join(graph_path, "graph/graph_ctc_3-gram.pruned.3e-7/TLG.fst"),
Expand Down
2 changes: 1 addition & 1 deletion third_party/kaldi

0 comments on commit 304b54a

Please sign in to comment.