Update bert_align.py

This commit is contained in:
nlpfun
2021-11-30 21:22:49 +08:00
parent 9636ea6966
commit 29d9d4ab24

View File

@@ -24,7 +24,7 @@ import numpy as np
import numba as nb import numba as nb
def main(): def main():
# user-defined parameters # User-defined parameters.
parser = argparse.ArgumentParser('Sentence alignment using Bertalign') parser = argparse.ArgumentParser('Sentence alignment using Bertalign')
parser.add_argument('-s', '--src', type=str, required=True, help='Source texts directory.') parser.add_argument('-s', '--src', type=str, required=True, help='Source texts directory.')
parser.add_argument('-t', '--tgt', type=str, required=True, help='Target texts directory.') parser.add_argument('-t', '--tgt', type=str, required=True, help='Target texts directory.')