Skip to main content
costa-huang
Projects
cleanRL
Reports
dm haiku vs flax
Log in
Sign up
Share
Comment
Star
Share
Comment
Star
dm haiku vs flax
Costa
Created on December 24
|
Last edited on December 25
Comment
system/gpu.process.1.gpu
system/gpu.process.1.gpu
2
4
6
8
10
Time (hours)
0
20
40
60
80
Alien-v5__ppo_atari_envpool_xla_jax_truncation_machado_impalanet_haiku__1__1671900014
dm-haiku
Alien-v5__ppo_atari_envpool_async_jax_impala__1__1671937140
flax async
Alien-v5__ppo_atari_envpool_xla_jax_truncation_machado_impalanet_haiku_scan__1__1671901265
dm-haiku scan
Alien-v5__ppo_atari_envpool_xla_jax_truncation_machado_impalanet__1__1671425456
flax
charts/SPS_update
charts/SPS_update
10M
20M
30M
40M
global_step
200
400
600
800
1000
1200
1400
charts/episodic_return, charts/avg_episodic_return
charts/episodic_return, charts/avg_episodic_return
2
4
6
8
10
Time (hours)
0
1000
2000
3000
4000
5000
6000
flax
1
dm-haiku
1
torch
1
dm-haiku scan
1
flax async
1
Add a comment