Skip to content

Commit 17548d5

Browse files
authored
Update link to image in Fabric PPO example (Lightning-AI#17360)
1 parent f14ee9e commit 17548d5

File tree

1 file changed

+1
-1
lines changed
  • examples/fabric/reinforcement_learning

1 file changed

+1
-1
lines changed

examples/fabric/reinforcement_learning/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ In this example we have gone even further leveraging the flexibility offered by
9191
The architecture is depicted in the following figure:
9292

9393
<p align="center">
94-
<img src="./imgs/fabric_decoupled.png">
94+
<img src="https://pl-public-data.s3.amazonaws.com/assets_lightning/examples/fabric/reinforcement-learning/ppo_fabric_decoupled.png">
9595
</p>
9696

9797
where, differently from the previous example, we have completely decoupled the environment from the agents: the *rank-0* process will be regarded as the *Player*, which runs `M+1` independent copies of the environment (*Env-0*, ..., *Env-M*); the *rank-1*, ..., *rank-N* are the *Trainers*, which contain the agent to be optimized. Player and Trainer share data through [collectives](https://lightning.ai/docs/fabric/stable/api/generated/lightning.fabric.plugins.collectives.TorchCollective.html#lightning.fabric.plugins.collectives.TorchCollective) and thanks to Fabric's flexibility we can run Player and Trainers on different devices.

0 commit comments

Comments
 (0)