Commit 37b54cb4 authored by williamzhangNU's avatar williamzhangNU
Browse files

update examples

parent 203675e4
Loading
Loading
Loading
Loading
+7 −4
Original line number Diff line number Diff line
@@ -71,21 +71,24 @@ We evaluated VAGEN on the visual puzzle-solving Sokoban task, demonstrating sign
The visualizations below show how the agent reasons through sequential steps to solve Sokoban puzzles, cherry picked from validation steps when training TRICO: LM + GM + Turn-wise Reward + Bi-Level GAE.

<p align="center">
    <img src="./public/example_1.png" width="1000px" alt="Example 1" />
    <img src="./public/example_4.png" width="400px" alt="Example 4: Masked GAE + Masked Loss" />
</p>

<p align="center">
    <img src="./public/example_2.png" width="1000px" alt="Example 2" />
    <img src="./public/example_3.png" width="600px" alt="Example 3: Masked GAE + Masked Loss" />
</p>

<p align="center">
    <img src="./public/example_3.png" width="1000px" alt="Example 3: Masked GAE + Masked Loss" />
    <img src="./public/example_1.png" width="1000px" alt="Example 1" />
</p>

<p align="center">
    <img src="./public/example_4.png" width="1000px" alt="Example 4: Masked GAE + Masked Loss" />
    <img src="./public/example_2.png" width="1000px" alt="Example 2" />
</p>




## Installation

```bash