From 194f04826388baf90ff757e2b3c26ba2e770b9ab Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sat, 13 Feb 2021 14:41:38 -0500
Subject: [PATCH] Update README.md

---
 README.md | 19 ++++---------------
 1 file changed, 4 insertions(+), 15 deletions(-)

diff --git a/README.md b/README.md
index e09c1f25d1..cdb0b402ff 100644
--- a/README.md
+++ b/README.md
@@ -213,9 +213,12 @@ Lightning has over [40+ advanced features](https://pytorch-lightning.readthedocs
 
 Here are some examples:
 
+<div align="center">
+  <img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/features_2.jpg" max-height="600px">
+</div>
 
 <details>
-  <summary>Train on GPUs without code changes</summary>
+  <summary>Highlighted feature code snippets</summary>
   
   ```python
   # 8 GPUs
@@ -225,27 +228,21 @@ Here are some examples:
   # 256 GPUs
   trainer = Trainer(max_epochs=1, gpus=8, num_nodes=32)
   ```
-</details>
 
-<details>
   <summary>Train on TPUs without code changes</summary>
   
   ```python
   # no code changes needed
   trainer = Trainer(tpu_cores=8)
    ```
-</details>
 
-<details>
   <summary>16-bit precision</summary>
   
   ```python
   # no code changes needed
   trainer = Trainer(precision=16)
    ```
-</details>
 
-<details>
   <summary>Experiment managers</summary>
   
   ```python
@@ -268,27 +265,21 @@ Here are some examples:
   
   # ... and dozens more
    ```
-</details>
 
-<details>
   <summary>EarlyStopping</summary>
   
   ```python
   es = EarlyStopping(monitor='val_loss')
   trainer = Trainer(callbacks=[es])
    ```
-</details>
 
-<details>
   <summary>Checkpointing</summary>
   
   ```python
   checkpointing = ModelCheckpoint(monitor='val_loss')
   trainer = Trainer(callbacks=[checkpointing])
    ```
-</details>
 
-<details>
   <summary>Export to torchscript (JIT) (production use)</summary>
   
   ```python
@@ -296,9 +287,7 @@ Here are some examples:
   autoencoder = LitAutoEncoder()
   torch.jit.save(autoencoder.to_torchscript(), "model.pt")
    ```
-</details>
 
-<details>
   <summary>Export to ONNX (production use)</summary>
   
   ```python