Error while testing

[hxqstree]

Thanks for your nice work!
I got the following error when testing:
Traceback (most recent call last):
File "test.py", line 93, in
main(args)
File "test.py", line 63, in main
test_ap = test(net, dataloader)
File "/home/hxq/HXQ/ViPLO/ViPLO-main/ViPLO/utils.py", line 239, in test
output = net(*inputs)
File "/home/hxq/anaconda3/envs/hoi/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/hxq/anaconda3/envs/hoi/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/hxq/HXQ/ViPLO/ViPLO-main/ViPLO/models.py", line 180, in forward
results = self.interaction_head(features, detections, image_sizes, targets, images)
File "/home/hxq/anaconda3/envs/hoi/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/hxq/anaconda3/envs/hoi/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/hxq/HXQ/ViPLO/ViPLO-main/ViPLO/interaction_head.py", line 362, in forward
box_features, global_features, patch_features = self.backbone.encode_image(images, box_coords, box_segs, need_patch=True)
TypeError: encode_image() got an unexpected keyword argument 'need_patch'