Fix typos. (#115)

This commit is contained in:
Fangjun Kuang 2021-11-10 14:45:30 +08:00 committed by GitHub
parent 21096e99d8
commit 8d679c3e74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions

View File

@ -60,14 +60,14 @@ def get_parser():
parser.add_argument(
"--epoch",
type=int,
default=34,
default=77,
help="It specifies the checkpoint to use for decoding."
"Note: Epoch counts from 0.",
)
parser.add_argument(
"--avg",
type=int,
default=20,
default=55,
help="Number of checkpoints to average. Automatically select "
"consecutive checkpoints before the checkpoint specified by "
"'--epoch'. ",

View File

@ -412,7 +412,9 @@ class Nbest(object):
scores_shape = self.fsa.arcs.shape().remove_axis(1)
# 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()