Potential bug in predict.py

namisan / mt-dnn

Multi-Task Deep Neural Networks for Natural Language Understanding

MIT License

2.24k stars 412 forks source link

Closed chawannut157 closed 2 years ago

chawannut157 commented 3 years ago

In predict.py, it's still using use_cuda=args.cuda, which is no longer used in the eval_model function.

eval_model(model, test_data,metric_meta=metric_meta,**use_cuda=args.cuda,** with_label=args.with_label)

In inference.py

def eval_model(model, data, metric_meta, device, with_label=True, label_mapper=None, task_type=TaskType.Classification):

namisan commented 2 years ago

thanks @chawannut157 , fixed the bug.