test.py have a error

[xiaomujiang]

when i run train.py , it have a error , but i do nothing to the code, can you help me ?

test.py", line 104, in test inf_out, train_out = model(img,augment = augment) # inference and training outputs ValueError: too many values to unpack (expected 2)