Can you support torch>1.7.0

selflein / GraphNN-Multi-Object-Tracking

Unofficial PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking"

122 stars 23 forks source link

Can you support torch>1.7.0 #12

Closed hahabooy closed 2 years ago

hahabooy commented 3 years ago

Hello, I found that pytorch-geometric is extremely difficult to install for torch1.3.0 as pytorch-geometric only supports torch>1.4.0 for now.

I think there won't be many changes to adapt current codes to torch>1.7.0. I'm new to GCN and torch-geometric and I'm willing to help.

selflein commented 3 years ago

Hi, I updated PyTorch to 1.8.2 and to pytorch-geometric 1.7.3 on the update_dependencies branch.

Hope this helps. I did not yet try to reproduce results.

quoccuonglqd commented 2 years ago

Hi selflein, thank you for your awesome project. I have tested the new branch with the inference.py. It seems like that a sub-module name node_encoder has been added to the the new definition of GNN module. But the pretrained model provided in the README.md do not contain this specific module. Could you pleased provide the corresponding pretrained model for this new branch? Thank you

selflein commented 2 years ago

I updated the model, updated to newer versions of PyTorch and PyTorch Geometric and merged the PR. So should work now. Feel free to re-open the issue if it is not resolved.