error when run the pose estimation example

[sundyCoder]

$ python3.6 pose_estimation_quanteval.py pe_weights.pth ./data/

2022-05-24 22:37:22,500 - root - INFO - AIMET
defining network with shared weights
Traceback (most recent call last):
File "pose_estimation_quanteval.py", line 700, in
pose_estimation_quanteval(args)
File "pose_estimation_quanteval.py", line 687, in pose_estimation_quanteval
sim = quantsim.QuantizationSimModel(model, dummy_input=(1, 3, 128, 128), quant_scheme=args.quant_scheme)
File "/home/jlchen/.local/lib/python3.6/site-packages/aimet_torch/quantsim.py", line 157, in init
self.connected_graph = ConnectedGraph(self.model, dummy_input)
File "/home/jlchen/.local/lib/python3.6/site-packages/aimet_torch/meta/connectedgraph.py", line 132, in init
self._construct_graph(model, model_input)
File "/home/jlchen/.local/lib/python3.6/site-packages/aimet_torch/meta/connectedgraph.py", line 254, in _construct_graph
module_tensor_shapes_map = ConnectedGraph._generate_module_tensor_shapes_lookup_table(model, model_input)
File "/home/jlchen/.local/lib/python3.6/site-packages/aimet_torch/meta/connectedgraph.py", line 244, in _generate_module_tensor_shapes_lookup_table
run_hook_for_layers_with_given_input(model, model_input, forward_hook, leaf_node_only=False)
File "/home/jlchen/.local/lib/python3.6/site-packages/aimet_torch/utils.py", line 277, in run_hook_for_layers_with_given_input
_ = model(*input_tensor)
File "/home/jlchen/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1071, in _call_impl
result = forward_call(*input, **kwargs)
TypeError: forward() takes 2 positional arguments but 5 were given