Update bert_align.py

This commit is contained in:
bfsujason
2021-11-28 19:18:35 +08:00
committed by GitHub
parent cfd11a65c3
commit 7eaeee4eb0

View File

@@ -26,7 +26,7 @@ import numba as nb
def main(): def main():
# user-defined parameters # user-defined parameters
parser = argparse.ArgumentParser('Sentence alignment using Vecalign') parser = argparse.ArgumentParser('Sentence alignment using Bertalign')
parser.add_argument('-s', '--src', type=str, required=True, help='preprocessed source file to align') parser.add_argument('-s', '--src', type=str, required=True, help='preprocessed source file to align')
parser.add_argument('-t', '--tgt', type=str, required=True, help='preprocessed target file to align') parser.add_argument('-t', '--tgt', type=str, required=True, help='preprocessed target file to align')
parser.add_argument('-o', '--out', type=str, required=True, help='Output directory.') parser.add_argument('-o', '--out', type=str, required=True, help='Output directory.')