issues
search
world-modelz
/
dreamax
A scalable Dreamer implementation in JAX
MIT License
11
stars
2
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Restore overall base training performance of April 7 repo state (before refactoring)
#20
andreaskoepf
opened
2 years ago
2
Fix `WARNING:root:NaN or Inf found in input tensor.`
#19
andreaskoepf
opened
2 years ago
0
Set new log dir, if no one is specified
#18
XMaster96
closed
2 years ago
1
Fix wrong color channel order of videos/env_render video
#17
andreaskoepf
opened
2 years ago
0
Add Training Integration Test
#16
XMaster96
opened
2 years ago
0
Use `jax.tree_util.tree_map()` instead of `jax.tree_util.tree_multimap()`
#15
andreaskoepf
opened
2 years ago
0
Bug: A function transformed by JAX had a side effect ...
#14
andreaskoepf
closed
2 years ago
3
Get rid of 'WARNING:root:The use of `check_types` is deprecated and does not have any effect.`
#13
andreaskoepf
opened
2 years ago
0
Implement Dreamer V2
#12
andreaskoepf
opened
2 years ago
0
Improve serialization of model & resumption of training
#11
andreaskoepf
opened
2 years ago
0
Add support for train/eval on Atari gym games (ALE)
#10
andreaskoepf
opened
2 years ago
0
Add logging to WnB as an option
#9
andreaskoepf
opened
2 years ago
0
Create an efficient replay buffer implementation (with serialization support)
#8
andreaskoepf
opened
2 years ago
0
Update readme for TPU VM usage instructions
#7
XMaster96
opened
2 years ago
0
TPU VM setup script
#6
XMaster96
opened
2 years ago
0
Data Parallelism on multiple accelerator devices
#5
XMaster96
opened
2 years ago
0
Parallel Rollout Worker
#4
XMaster96
opened
2 years ago
0
Master Model Parameter copy
#3
XMaster96
closed
2 years ago
0
inference world model instants
#2
XMaster96
closed
2 years ago
0
CuDNN related warnings
#1
andreaskoepf
opened
2 years ago
1