JAX PPO + Atari

Created on July 6|Last edited on July 9
Comment
﻿
﻿
losses/approx_kl
losses/approx_kl
2M4M6M8Mglobal_step0.0050.010.0150.02
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
ppo_atari_envpool_xla_jax.py (no clip_vloss)
losses/value_loss
losses/value_loss
2M4M6M8Mglobal_step0.20.40.60.8
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
ppo_atari_envpool_xla_jax.py (no clip_vloss)
charts/episodic_length
charts/episodic_length
2M4M6M8Mglobal_step200040006000800010000
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
charts/SPS
charts/SPS
2M4M6M8Mglobal_step1000200030004000
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
ppo_atari_envpool_xla_jax.py (no clip_vloss)
Breakout
Breakout
2M4M6M8MStep100200300400Episodic Return
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
ppo_atari_envpool_xla_jax.py (no clip_vloss)
charts/avg_episodic_return
charts/avg_episodic_return
20406080Time (minutes)0100200300400
ppo_atari_envpool.py
ppo_atari_envpool.py (no clip_vloss)
ppo_atari_envpool_jax.py (no clip_vloss)
ppo_atari_envpool_xla_jax.py (no clip_vloss)
ppo_atari_envpool.py3
ppo_atari_envpool.py (no clip_vloss)2
ppo_atari_envpool_jax.py (no clip_vloss)1
ppo_atari_envpool_xla_jax.py (no clip_vloss)1
 
a2c1
﻿
﻿
﻿
ppo_atari_envpool_jax.py1
ppo_atari_envpool.py1
ppo_atari_envpool_xla_jax.py1
﻿
﻿
Add a comment