Zhixin's workspace
1
meta
["--seed","0","--n_envs","16","--n_steps","128","--learning_rate","0.0001","--batch-size","64","--total_timesteps","1000000"]
["--sbx","--device","cuda","--seed","0","--n_envs","16","--n_steps","128","--learning_rate","0.0001","--batch-size","64","--total_timesteps","1000000"]
["--sbx","--seed","0","--n_envs","16","--n_steps","128","--learning_rate","0.0001","--batch-size","64","--total_timesteps","1000000"]
"brax_learn.py"
"sb3_learn.py"
"sb3_learn.py"
/home/zhixin/jax_learn/brax_learn.py
/home/zhixin/jax_learn/sb3_learn.py
/home/zhixin/jax_learn/sb3_learn.py
6m 34s
3m 42s
3m 45s
config
cuda
cuda
cpu
summary
_wandb
392
219
223
rollout
450.69
544.05
532.81
1719.1929
1943.7355
1813.9386
time
2660
4647
4558
train
0.99546
0.98137
0.99424
-0.033877
-0.0035437
-0.058106
0.54477
0.61121
0.64724
8.98099
18.66479
26.23045
1
2
1
5