├── .gitignore ├── AUTHORS ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── dreamer ├── __init__.py ├── control │ ├── __init__.py │ ├── batch_env.py │ ├── dummy_env.py │ ├── mpc_agent.py │ ├── planning.py │ ├── random_episodes.py │ ├── simulate.py │ ├── temporal_difference.py │ └── wrappers.py ├── distributions │ ├── __init__.py │ ├── one_hot.py │ └── tanh_normal.py ├── models │ ├── __init__.py │ ├── base.py │ └── rssm.py ├── networks │ ├── __init__.py │ ├── basic.py │ ├── conv.py │ └── proprio.py ├── scripts │ ├── __init__.py │ ├── configs.py │ ├── fileutil_fetch.py │ ├── objectives.py │ ├── sync.py │ ├── tasks.py │ └── train.py ├── tools │ ├── __init__.py │ ├── attr_dict.py │ ├── bind.py │ ├── chunk_sequence.py │ ├── copy_weights.py │ ├── count_dataset.py │ ├── count_weights.py │ ├── custom_optimizer.py │ ├── filter_variables_lib.py │ ├── gif_summary.py │ ├── image_strip_summary.py │ ├── mask.py │ ├── metrics.py │ ├── nested.py │ ├── numpy_episodes.py │ ├── overshooting.py │ ├── preprocess.py │ ├── reshape_as.py │ ├── schedule.py │ ├── shape.py │ ├── streaming_mean.py │ ├── summary.py │ ├── target_network.py │ └── unroll.py └── training │ ├── __init__.py │ ├── define_model.py │ ├── define_summaries.py │ ├── running.py │ ├── trainer.py │ └── utility.py ├── setup.py └── tests ├── test_dreamer.py ├── test_isolate_envs.py ├── test_nested.py ├── test_overshooting.py └── test_running.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | *.py[cod] 3 | *.egg-info 4 | /dist 5 | /logdir 6 | MUJOCO_LOG.TXT 7 | -------------------------------------------------------------------------------- /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/AUTHORS -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/README.md -------------------------------------------------------------------------------- /dreamer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/__init__.py -------------------------------------------------------------------------------- /dreamer/control/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/__init__.py -------------------------------------------------------------------------------- /dreamer/control/batch_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/batch_env.py -------------------------------------------------------------------------------- /dreamer/control/dummy_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/dummy_env.py -------------------------------------------------------------------------------- /dreamer/control/mpc_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/mpc_agent.py -------------------------------------------------------------------------------- /dreamer/control/planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/planning.py -------------------------------------------------------------------------------- /dreamer/control/random_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/random_episodes.py -------------------------------------------------------------------------------- /dreamer/control/simulate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/simulate.py -------------------------------------------------------------------------------- /dreamer/control/temporal_difference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/temporal_difference.py -------------------------------------------------------------------------------- /dreamer/control/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/control/wrappers.py -------------------------------------------------------------------------------- /dreamer/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/distributions/__init__.py -------------------------------------------------------------------------------- /dreamer/distributions/one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/distributions/one_hot.py -------------------------------------------------------------------------------- /dreamer/distributions/tanh_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/distributions/tanh_normal.py -------------------------------------------------------------------------------- /dreamer/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/models/__init__.py -------------------------------------------------------------------------------- /dreamer/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/models/base.py -------------------------------------------------------------------------------- /dreamer/models/rssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/models/rssm.py -------------------------------------------------------------------------------- /dreamer/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/networks/__init__.py -------------------------------------------------------------------------------- /dreamer/networks/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/networks/basic.py -------------------------------------------------------------------------------- /dreamer/networks/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/networks/conv.py -------------------------------------------------------------------------------- /dreamer/networks/proprio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/networks/proprio.py -------------------------------------------------------------------------------- /dreamer/scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/__init__.py -------------------------------------------------------------------------------- /dreamer/scripts/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/configs.py -------------------------------------------------------------------------------- /dreamer/scripts/fileutil_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/fileutil_fetch.py -------------------------------------------------------------------------------- /dreamer/scripts/objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/objectives.py -------------------------------------------------------------------------------- /dreamer/scripts/sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/sync.py -------------------------------------------------------------------------------- /dreamer/scripts/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/tasks.py -------------------------------------------------------------------------------- /dreamer/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/scripts/train.py -------------------------------------------------------------------------------- /dreamer/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/__init__.py -------------------------------------------------------------------------------- /dreamer/tools/attr_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/attr_dict.py -------------------------------------------------------------------------------- /dreamer/tools/bind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/bind.py -------------------------------------------------------------------------------- /dreamer/tools/chunk_sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/chunk_sequence.py -------------------------------------------------------------------------------- /dreamer/tools/copy_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/copy_weights.py -------------------------------------------------------------------------------- /dreamer/tools/count_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/count_dataset.py -------------------------------------------------------------------------------- /dreamer/tools/count_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/count_weights.py -------------------------------------------------------------------------------- /dreamer/tools/custom_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/custom_optimizer.py -------------------------------------------------------------------------------- /dreamer/tools/filter_variables_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/filter_variables_lib.py -------------------------------------------------------------------------------- /dreamer/tools/gif_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/gif_summary.py -------------------------------------------------------------------------------- /dreamer/tools/image_strip_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/image_strip_summary.py -------------------------------------------------------------------------------- /dreamer/tools/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/mask.py -------------------------------------------------------------------------------- /dreamer/tools/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/metrics.py -------------------------------------------------------------------------------- /dreamer/tools/nested.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/nested.py -------------------------------------------------------------------------------- /dreamer/tools/numpy_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/numpy_episodes.py -------------------------------------------------------------------------------- /dreamer/tools/overshooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/overshooting.py -------------------------------------------------------------------------------- /dreamer/tools/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/preprocess.py -------------------------------------------------------------------------------- /dreamer/tools/reshape_as.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/reshape_as.py -------------------------------------------------------------------------------- /dreamer/tools/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/schedule.py -------------------------------------------------------------------------------- /dreamer/tools/shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/shape.py -------------------------------------------------------------------------------- /dreamer/tools/streaming_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/streaming_mean.py -------------------------------------------------------------------------------- /dreamer/tools/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/summary.py -------------------------------------------------------------------------------- /dreamer/tools/target_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/target_network.py -------------------------------------------------------------------------------- /dreamer/tools/unroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/tools/unroll.py -------------------------------------------------------------------------------- /dreamer/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/__init__.py -------------------------------------------------------------------------------- /dreamer/training/define_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/define_model.py -------------------------------------------------------------------------------- /dreamer/training/define_summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/define_summaries.py -------------------------------------------------------------------------------- /dreamer/training/running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/running.py -------------------------------------------------------------------------------- /dreamer/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/trainer.py -------------------------------------------------------------------------------- /dreamer/training/utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/dreamer/training/utility.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_dreamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/tests/test_dreamer.py -------------------------------------------------------------------------------- /tests/test_isolate_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/tests/test_isolate_envs.py -------------------------------------------------------------------------------- /tests/test_nested.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/tests/test_nested.py -------------------------------------------------------------------------------- /tests/test_overshooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/tests/test_overshooting.py -------------------------------------------------------------------------------- /tests/test_running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/dreamer/HEAD/tests/test_running.py --------------------------------------------------------------------------------