THUNLP-MT/THUMT

Code Problem and Potential Solution: Inference with CPU

Beau-xu opened this issue · 0 comments

Beau-xu commented

It seems that the parameter enabling CPU inference is not correctly passed to the function get_infer_dataset. Actually, the function get_eval_dataset has the same issue.

image

Those who have this problem can edit the code in /bin/translator.py and append the parameter cpu=args.cpu

data.MTPipeline.get_infer_dataset(args.input[0], params, cpu=args.cpu)