diff --git a/fairseq/hub_utils.py b/fairseq/hub_utils.py index 1b1091c881..c249eb23f5 100644 --- a/fairseq/hub_utils.py +++ b/fairseq/hub_utils.py @@ -189,7 +189,7 @@ def getarg(name, default): )) if hypo['alignment'] is not None and getarg('print_alignment', False): logger.info('A\t{}'.format( - ' '.join(map(lambda x: str(utils.item(x)), hypo['alignment'].int().cpu())) + ' '.join(['{}-{}'.format(src_idx, tgt_idx) for src_idx, tgt_idx in hypo['alignment']]) )) return outputs