search

selflein / GraphNN-Multi-Object-Tracking

Unofficial PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking"
121 stars 23 forks source link

Error:AttributeError: 'TimeDependent' object has no attribute '__message_args__' #6

Closed xiaocc612 closed 4 years ago

xiaocc612 commented 4 years ago

When I run train.py or inference.py，message_passing.py will be executed. The error occurs.

Traceback (most recent call last): File "inference.py", line 152, in <module> Path(args.net_weights)) File "inference.py", line 126, in get_track_dict device) File "inference.py", line 53, in combine_subsequences pred = net(data, node_features.clone().to(device)).cpu().numpy() File "~/anaconda3/envs/gnn_mot/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__ result = self.forward(*input, **kwargs) File "../gnn_tracker/gnn_modules/graph_nn.py", line 45, in forward x, edge_attr = self.td1(x, edge_index, edge_attr, node_timestamps, initial_x) File "~/anaconda3/envs/gnn_mot/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__ result = self.forward(*input, **kwargs) File "../gnn_tracker/gnn_modules/message_passing.py", line 42, in forward initial_x=initial_x) File "../gnn_tracker/gnn_modules/message_passing.py", line 82, in propagate for arg in self.__message_args__: File "/~/anaconda3/envs/gnn_mot/lib/python3.7/site-packages/torch/nn/modules/module.py", line 576, in __getattr__ type(self).__name__, name)) AttributeError: 'TimeDependent' object has no attribute '__message_args__' I don't find the attribute __message_args__ in torch_geometric.nn.MessagePassing nor torch.nn.modules.module. Can you help me to solve this? Thanks.

selflein commented 4 years ago

https://github.com/selflein/GraphNN-Multi-Object-Tracking/issues/2 has the fix. Please use pytorch-geometric 1.3.2

xiaocc612 commented 4 years ago

2 has the fix. Please use pytorch-geometric 1.3.2

Thanks : )