yolov5训练生成了best.pt文件，使用test.py文件测试时出现问题

yolov5训练生成了best.pt文件，使用test.py文件测试时出现问题

Traceback (most recent call last):
  File "test.py", line 334, in <module>
    save_conf=opt.save_conf,
  File "test.py", line 89, in test
    dataloader = create_dataloader(data[task], imgsz, batch_size, gs, opt, pad=0.5, rect=True,
KeyError: 'test'

已解决，因为data里面yaml文件没有给test的路径