diff --git a/agent/sequence_generator.py b/agent/sequence_generator.py index e364eaf..7b2a914 100644 --- a/agent/sequence_generator.py +++ b/agent/sequence_generator.py @@ -282,7 +282,7 @@ def generate_decoder( finalized = torch.jit.annotate( List[List[Dict[str, Tensor]]], [torch.jit.annotate(List[Dict[str, Tensor]], []) for i in range(bsz)], - ) # contains lists of dictionaries of infomation about the hypothesis being finalized at each step + ) # contains lists of dictionaries of information about the hypothesis being finalized at each step # a boolean array indicating if the sentence at the index is finished or not finished = [False for i in range(bsz)]