Skip to content
Snippets Groups Projects
Unverified Commit 30cf0706 authored by Aditya Prakash's avatar Aditya Prakash Committed by GitHub
Browse files

Update README.md

parent 7df4fe18
No related branches found
No related tags found
No related merge requests found
# AIM
![aim](assets/model.png)
# CILRS
<p align="center"> <img src="assets/model.png" width="512"> </p>
AIM consists of a ResNet34 image encoder with an autoregressive GRU-based waypoint prediction network. This is equivalent to adapting CILRS to predict waypoints conditioned on goal locations rather than predicting vehicle controls conditioned on navigational commmands.
## Training
```Shell
CUDA_VISIBLE_DEVICES=<gpu_id> python3 train.py --id cilrs --batch_size 256
```
## Evaluation
Update ```leaderboard/scripts/run_evaluation.sh``` to include the following.
```
export ROUTES=leaderboard/data/evaluation_routes/routes_town05_long.xml
export TEAM_AGENT=leaderboard/team_code/cilrs_agent.py
export TEAM_CONFIG=model_ckpt/cilrs
export CHECKPOINT_ENDPOINT=results/cilrs_result.json
export SCENARIOS=leaderboard/data/scenarios/town05_all_scenarios.json
```
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment