search

selflein / GraphNN-Multi-Object-Tracking

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

Some code does not match #2

Closed derkbreeze closed 4 years ago

derkbreeze commented 4 years ago

https://github.com/selflein/GraphNN-Multi-Object-Tracking/blob/22108414124871908419727488c92406ebf8e7ea/src/gnn_tracker/modules/message_passing.py#L123 self.message_args self.update_argsboth are not found. I'm using torch geometric 1.4.3 version, is it because version mismatch? What could be the suggestions to modify this, thank you

derkbreeze commented 4 years ago

Mainly because of version mismatch. Problem solved. Solution: import inspect self.__message_args__ = inspect.getargspec(self.message) where self is an object of message passing class.