mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 10:02:22 +00:00
Fix typos. (#115)
This commit is contained in:
parent
21096e99d8
commit
8d679c3e74
@ -60,14 +60,14 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--epoch",
|
"--epoch",
|
||||||
type=int,
|
type=int,
|
||||||
default=34,
|
default=77,
|
||||||
help="It specifies the checkpoint to use for decoding."
|
help="It specifies the checkpoint to use for decoding."
|
||||||
"Note: Epoch counts from 0.",
|
"Note: Epoch counts from 0.",
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--avg",
|
"--avg",
|
||||||
type=int,
|
type=int,
|
||||||
default=20,
|
default=55,
|
||||||
help="Number of checkpoints to average. Automatically select "
|
help="Number of checkpoints to average. Automatically select "
|
||||||
"consecutive checkpoints before the checkpoint specified by "
|
"consecutive checkpoints before the checkpoint specified by "
|
||||||
"'--epoch'. ",
|
"'--epoch'. ",
|
||||||
|
@ -412,7 +412,9 @@ class Nbest(object):
|
|||||||
scores_shape = self.fsa.arcs.shape().remove_axis(1)
|
scores_shape = self.fsa.arcs.shape().remove_axis(1)
|
||||||
# scores_shape has axes [path][arc]
|
# scores_shape has axes [path][arc]
|
||||||
|
|
||||||
ragged_scores = k2.RaggedTensor(scores_shape, self.scores.contiguous())
|
ragged_scores = k2.RaggedTensor(
|
||||||
|
scores_shape, self.fsa.scores.contiguous()
|
||||||
|
)
|
||||||
|
|
||||||
tot_scores = ragged_scores.sum()
|
tot_scores = ragged_scores.sum()
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user