import argparse
import os
from parser.cmds import Evaluate, Predict, Train
from parser.config import Config
import torch
if __name__ == '__main__':
parser = argparse.ArgumentParser(
description='Create the Biaffine Parser model.'
)
subparsers = parser.add_subparsers(title='Commands', dest='mode')
subcommands = {
'evaluate': Evaluate(),
'predict': Predict(),
'train': Train()
}
for name, subcommand in subcommands.items():
subparser = subcommand.add_subparser(name, subparsers)
subparser.add_argument('--conf', '-c', default='config.ini',
help='path to config file')
subparser.add_argument('--file', '-f', default='exp/ptb',
help='path to saved files')
subparser.add_argument('--preprocess', '-p', action='store_true',
help='whether to preprocess the data first')
subparser.add_argument('--device', '-d', default='-1',
help='ID of GPU to use')
subparser.add_argument('--seed', '-s', default=1, type=int,
help='seed for generating random numbers')
subparser.add_argument('--threads', '-t', default=16, type=int,
# -*- coding: utf-8 -*-
import argparse
import os
from parser.cmds import Predict, Train
import torch
from config import Config
if __name__ == '__main__':
parser = argparse.ArgumentParser(
description='Create the Biaffine Parser model.')
subparsers = parser.add_subparsers(title='Commands', dest='mode')
subcommands = {'predict': Predict(), 'train': Train()}
for name, subcommand in subcommands.items():
subparser = subcommand.add_subparser(name, subparsers)
subparser.add_argument('--conf',
'-c',
default='config.ini',
help='path to config file')
subparser.add_argument('--model',
'-m',
default='exp/',
help='path to model file')
subparser.add_argument('--vocab',
'-v',
default='exp/',
help='path to vocab file')
subparser.add_argument('--device',
'-d',
import os
import pathlib
from parser.cmds import Evaluate, Predict, Train
from parser.config import Config
from parser.utils.os_utils import path_or_name
from parser.utils.gpu_utils import get_gpus_configuration
import torch
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Create the Biaffine Parser model.")
subparsers = parser.add_subparsers(title="Commands", dest="mode")
subcommands = {
"evaluate": Evaluate(),
"predict": Predict(),
"train": Train()
}
for name, subcommand in subcommands.items():
print(name, subcommand)
subparser = subcommand.add_subparser(name, subparsers)
subparser.add_argument("--conf",
"-c",
default="config.ini",
help="path to config file")
subparser.add_argument("--folder",
"-f",
required=True,
help="path to project folder")
subparser.add_argument("--model",
"-m",