poetry_diacritizer/test.py

import argparse
import random
from tester import DiacritizationTester

import numpy as np
import torch


SEED = 1234
random.seed(SEED)
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False


def train_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument("--model", dest="model_kind", type=str, required=True)
    parser.add_argument("--config", dest="config", type=str, required=True)
    parser.add_argument("--model_path", dest="model_path", type=str, required=False)
    parser.add_argument("--test", dest="test", type=bool)
    return parser


parser = train_parser()
args = parser.parse_args()

tester = DiacritizationTester(args.config, args.model_kind)
tester.run()