From 958f84a8e42745a065d44df3a6d6f80289e42667 Mon Sep 17 00:00:00 2001
From: Federico Berto <berto.federico2@gmail.com>
Date: Fri, 10 Nov 2023 15:31:29 +0900
Subject: [PATCH] [Docs] add images

---
 docs/_content/api/models/base.md           |  5 +++++
 docs/_content/api/models/env_embeddings.md | 11 +++++++++++
 docs/_content/api/models/zoo.md            | 10 ----------
 3 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/docs/_content/api/models/base.md b/docs/_content/api/models/base.md
index b90db27f..fecedf1a 100644
--- a/docs/_content/api/models/base.md
+++ b/docs/_content/api/models/base.md
@@ -1,5 +1,10 @@
 # Base Autoregressive Model
 
+
+<img class="full-img" alt="policy" src="https://user-images.githubusercontent.com/48984123/281976545-ca88f159-d0b3-459e-8fd9-89799be9d1b0.png">
+
+
+
 ## Policy
 
 ```{eval-rst}
diff --git a/docs/_content/api/models/env_embeddings.md b/docs/_content/api/models/env_embeddings.md
index 3749de42..9d70ae6b 100644
--- a/docs/_content/api/models/env_embeddings.md
+++ b/docs/_content/api/models/env_embeddings.md
@@ -1,5 +1,16 @@
 # Environment Embeddings
 
+In autoregressive policies, environment embeddings transfer data from feature space to hidden space:
+
+- Initial Embeddings: encode global problem features
+- Context Embeddings: modify current node embedding during decoding
+- Dynamic Embeddings: modify all nodes embeddings during decoding
+
+<img class="full-img" alt="policy" src="https://user-images.githubusercontent.com/48984123/281976545-ca88f159-d0b3-459e-8fd9-89799be9d1b0.png">
+
+
+
+
 ## Context Embeddings
 
 The context embedding is used to modify the query embedding of the problem node of the current partial solution. Usually consists of a projection of gathered node embeddings and features to the embedding space.
diff --git a/docs/_content/api/models/zoo.md b/docs/_content/api/models/zoo.md
index 362d8436..aa1de066 100644
--- a/docs/_content/api/models/zoo.md
+++ b/docs/_content/api/models/zoo.md
@@ -34,11 +34,6 @@ Models from the literature and contributions are contained in the Model Zoo.
    :undoc-members:
 ```
 
-```{eval-rst}
-.. automodule:: rl4co.models.zoo.ppo.decoder
-   :members:
-   :undoc-members:
-```
 
 ## Heterogeneous Attention Model (HAM)
 
@@ -134,11 +129,6 @@ Models from the literature and contributions are contained in the Model Zoo.
    :undoc-members:
 ```
 
-```{eval-rst}
-.. automodule:: rl4co.models.zoo.pomo.decoder
-   :members:
-   :undoc-members:
-```
 
 ## Pointer Network (PtrNet)