diff --git a/deploy/demo_server.py b/deploy/demo_server.py index 88703e5f600e3b30e5ff2d55930228d6e40c144e..d64f9f015513a021bfc3bc40e05900ace8b6b80e 100644 --- a/deploy/demo_server.py +++ b/deploy/demo_server.py @@ -23,8 +23,8 @@ add_arg('beam_size', int, 500, "Beam search width.") add_arg('num_conv_layers', int, 2, "# of convolution layers.") add_arg('num_rnn_layers', int, 3, "# of recurrent layers.") add_arg('rnn_layer_size', int, 2048, "# of recurrent cells per layer.") -add_arg('alpha', float, 2.15, "Coef of LM for beam search.") -add_arg('beta', float, 0.35, "Coef of WC for beam search.") +add_arg('alpha', float, 2.5, "Coef of LM for beam search.") +add_arg('beta', float, 0.3, "Coef of WC for beam search.") add_arg('cutoff_prob', float, 1.0, "Cutoff probability for pruning.") add_arg('cutoff_top_n', int, 40, "Cutoff number for pruning.") add_arg('use_gru', bool, False, "Use GRUs instead of simple RNNs.") diff --git a/infer.py b/infer.py index 7e30549ae0ec7a246f01cc8bd6e396a02d7e3cf8..b801c507b7269148c3ff5dc406520bb4f5113092 100644 --- a/infer.py +++ b/infer.py @@ -21,8 +21,8 @@ add_arg('num_proc_bsearch', int, 8, "# of CPUs for beam search.") add_arg('num_conv_layers', int, 2, "# of convolution layers.") add_arg('num_rnn_layers', int, 3, "# of recurrent layers.") add_arg('rnn_layer_size', int, 2048, "# of recurrent cells per layer.") -add_arg('alpha', float, 2.15, "Coef of LM for beam search.") -add_arg('beta', float, 0.35, "Coef of WC for beam search.") +add_arg('alpha', float, 2.5, "Coef of LM for beam search.") +add_arg('beta', float, 0.3, "Coef of WC for beam search.") add_arg('cutoff_prob', float, 1.0, "Cutoff probability for pruning.") add_arg('cutoff_top_n', int, 40, "Cutoff number for pruning.") add_arg('use_gru', bool, False, "Use GRUs instead of simple RNNs.") diff --git a/test.py b/test.py index 85b49f2ae1a546c5ce47e2ee2b661b35a42281aa..5cf7664870a1167cc4838e70e2e7213594a24f9f 100644 --- a/test.py +++ b/test.py @@ -22,8 +22,8 @@ add_arg('num_proc_data', int, 8, "# of CPUs for data preprocessing.") add_arg('num_conv_layers', int, 2, "# of convolution layers.") add_arg('num_rnn_layers', int, 3, "# of recurrent layers.") add_arg('rnn_layer_size', int, 2048, "# of recurrent cells per layer.") -add_arg('alpha', float, 2.15, "Coef of LM for beam search.") -add_arg('beta', float, 0.35, "Coef of WC for beam search.") +add_arg('alpha', float, 2.5, "Coef of LM for beam search.") +add_arg('beta', float, 0.3, "Coef of WC for beam search.") add_arg('cutoff_prob', float, 1.0, "Cutoff probability for pruning.") add_arg('cutoff_top_n', int, 40, "Cutoff number for pruning.") add_arg('use_gru', bool, False, "Use GRUs instead of simple RNNs.")