From 4c81ff806ec6d599e3dba2e17e4090ebe6e4e959 Mon Sep 17 00:00:00 2001
From: Edan Toledo <42650996+EdanToledo@users.noreply.github.com>
Date: Sun, 10 Mar 2024 23:48:42 +0000
Subject: [PATCH] Update README.md

---
 README.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 97296246..483fddb2 100644
--- a/README.md
+++ b/README.md
@@ -73,9 +73,9 @@ Stoix natively supports logging to json files which adhere to the standard sugge
 As the code in Stoix (at the time of creation) was in essense a port of [Mava][mava], for further speed comparisons we point to their repo. Additionally, we refer to the PureJaxRL blog post [here](https://chrislu.page/blog/meta-disco/) where the speed benefits of end-to-end JAX systems are discussed.
 
 Below we provide some plots illustrating that Stoix performs equally to that of [PureJaxRL][purejaxrl] but with the added benefit of the code being already set up for `pmap` distribution over devices as well as the other features provided (algorithm implementations, logging, config system, etc).
-
-<img src="docs/images/ppo_compare.png" alt="ppo" width="500"/> <img src="docs/images/dqn_compare.png" alt="dqn" width="500"/>
-
+<p align="center">
+<img src="docs/images/ppo_compare.png" alt="ppo" width="45%"/> <img src="docs/images/dqn_compare.png" alt="dqn" width="45%"/>
+</p>
 I've also included a plot of the training time for 5e5 steps of PPO as one scales the number of environments. PureJaxRL does not pmap and thus runs on a single a device.
 
 <p align="center">