KO
|
EN
gitlite — search
Search
#python
#java
#python3
#arduino
#golang
#machine-learning
#rust
#html
#flask
#javascript
#seismology
#nodejs
GRIN_NeurIPS21
★ 15
Open GitHub ↗
No description available.
Download README (.md)
Explore Similar Repositories
fairygui-phaser-example
:
fairygui demo
usad-on-ucr-data
:
USAD model on UCR Time Series Anomaly Archive
luci-app-socat
:
Web UI AT-commands using socat for OpenWrt.
awesome-object-slam
:
Personal paper collection for object SLAM
Kyo
:
No description available.
// repository documentation
Was this content helpful?
★ 0
(0 ratings)
Select Rating:
★
★
★
★
★
Submit Feedback
Recent Feedback
×
Download README
Do you want to download the
README.md
file for
GRIN_NeurIPS21
?
Download (.md)
# GRIN: Generative Relation and Intention Network for Multi-agent Trajectory Prediction Official Implementation of Generative Relation and Intention Network (GRIN) in PyTorch and DGL. ## Dependencies - torch==1.8.1 - numpy==1.19.2 - scipy==1.6.1 - dgl_cu110==0.6.1 - dgl==0.6.1 - tensorboardX==2.2 ## Running the code 0. Install all dependencies mentioned above 1. Generate charged dataset for training (NBA dataset is available on [44]) ``` python simulator.py --seed 0 --num_sample 5000 --filename train.npz python simulator.py --seed 1 --num_sample 1000 --filename test.npz python simulator.py --seed 2 --num_sample 1000 --filename valid.npz ``` 2. Train the model ``` bash train.sh ``` 3. Evaluate the model ``` bash eval.sh ```
