def _get_parser():
parser = ArgumentParser(description='train.py')
opts.config_opts(parser)
opts.model_opts(parser)
opts.train_opts(parser)
parser.add('--data', '-data', required=False,
default='F:/Project/Python/selfProject/translate_NMT/data/demo',
help='Path prefix to the ".train.pt" and '
'".valid.pt" file path from preprocess.py')
parser.add('--save_model', '-save_model', required=False,
default='F:/Project/Python/selfProject/translate_NMT/data',
help="Model filename (the model will be saved as "
"<save_model>_N.pt where N is the number "
"of steps")
parser.add('--save_checkpoint_steps', '-save_checkpoint_steps',
type=int, default=500,
help="""Save a checkpoint every X steps""")
parser.add('--train_from', '-train_from',
# default='F:/Project/Python/selfProject/translate_NMT/data/demo-model_step_150.pt',
default='',
type=str,
help="If training from a checkpoint then this is the "
"path to the pretrained model's state_dict.")
# default = 100000,
parser.add('--train_steps', '-train_steps', type=int, default=100000,
help='训练多少步')
return parser
def _get_parser():
parser = ArgumentParser(description='translate.py')
opts.config_opts(parser)
opts.translate_opts(parser)
parser.add(
'--model',
'-model',
dest='models',
metavar='MODEL',
nargs='+',
type=str,
default=[
"F:/Project/Python/selfProject/translate_NMT/transflate_NMT/data_step_100.pt"
],
required=False,
help="模型使用得训练文件")
parser.add(
'--src',
'-src',
required=False,
default=
"F:/Project/Python/selfProject/translate_NMT/transflate_NMT/data/src-test.txt",
help="自己写的测试文件在哪里？？？？")
parser.add(
'--output',
'-output',
default=
'F:/Project/Python/selfProject/translate_NMT/transflate_NMT/data/pred.txt',
help="测试文件输出未知 改成自己得")
return parser
def _get_parser():
parser = ArgumentParser(description='preprocess.py')
parser.add('--train_src', '-train_src', required=False, nargs='+',
default=['F:/Project/Python/selfProject/translate_NMT/data/src-train.txt'],
help="Path(s) to the training source data")
parser.add('--train_tgt', '-train_tgt', required=False, nargs='+',
default=['F:/Project/Python/selfProject/translate_NMT/data/tgt-train.txt'],
help="Path(s) to the training target data")
parser.add('--valid_src', '-valid_src',
default='F:/Project/Python/selfProject/translate_NMT/data/src-val.txt',
help="Path to the validation source data")
parser.add('--valid_tgt', '-valid_tgt',
default='F:/Project/Python/selfProject/translate_NMT/data/tgt-val.txt',
help="Path to the validation target data")
parser.add('--save_data', '-save_data', required=False,
default='F:/Project/Python/selfProject/translate_NMT/data/demo',
help="Output file for the prepared data")
parser.add('--num_threads', '-num_threads', type=int, default=1,
help="Number of shards to build in parallel.")
parser_init(parser)
return parser
