out, train_out = model(img, augment=augment) # inference and training outputs

[kailin zhang]

File "/home/code/yuhuankeji/yolov5-5.0/train.py", line 365, in train
is_coco=is_coco)
File "/homel/code/yuhuankeji/yolov5-5.0/test.py", line 110, in test
out, train_out = model(img, augment=augment) # inference and training outputs
ValueError: too many values to unpack (expected 2)

除了注释 yolo.py部分代码外 ，这个代码是不是也要修改啊