def main(args):
"""
main_func
"""
if args.print_config:
print_arguments(args)
start_procs(args)
if 'kn' in args.task_name:
args.use_knowledge = True
else:
args.use_knowledge = False
processor, [train_data, dev_data], warmup_steps = build_data(args)
args.voc_size = len(
open(args.vocab_path, 'r', encoding='utf-8').readlines())
num_labels = len(processor.get_labels())
checkpoint = None
if args.init_checkpoint:
print('Loading checkpoint from %s' % args.init_checkpoint)
checkpoint = torch.load(args.init_checkpoint,
map_location=lambda storage, loc: storage)
retrieval_model = build_model(args, num_labels, checkpoint)
optimizer = build_optim(retrieval_model, args, warmup_steps, checkpoint)
train_model(retrieval_model, optimizer, train_data, dev_data, processor,
args)
if __name__ == '__main__':
args = base_parser()
print_arguments(args)
main(args)