eval_pair.py

from utils import tab_printer
from sg_net import SGTrainer
from parser_sg import sgpr_args


def main():
    args = sgpr_args()
    args.load('./config/config.yml')
    tab_printer(args)
    trainer = SGTrainer(args, False)
    trainer.model.eval()
    pred, gt = trainer.eval_batch_pair([args.pair_file,])
    print("Score:",pred[0])


if __name__ == "__main__":
    main()