From 7df4fe182dcc93de7587a2e82266b98b09df10f7 Mon Sep 17 00:00:00 2001
From: Aditya Prakash <adityaprakash229997@gmail.com>
Date: Wed, 14 Apr 2021 10:59:02 +0200
Subject: [PATCH] Update README.md

---
 aim/README.md | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/aim/README.md b/aim/README.md
index 56ced29..ce0e0c9 100644
--- a/aim/README.md
+++ b/aim/README.md
@@ -1,2 +1,20 @@
 # AIM
-![aim](assets/model.png)
+
+<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 aim --batch_size 192
+```
+
+## 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/aim_agent.py
+export TEAM_CONFIG=model_ckpt/aim
+export CHECKPOINT_ENDPOINT=results/aim_result.json
+export SCENARIOS=leaderboard/data/scenarios/town05_all_scenarios.json
+```
-- 
GitLab