├── .gitignore ├── LICENSE ├── README.md ├── dreamer ├── algorithms │ ├── dreamer.py │ └── plan2explore.py ├── configs │ ├── atari-boxing-v4.yml │ ├── dmc-acrobot-swingup.yml │ ├── dmc-ball-in-cup-catch.yml │ ├── dmc-cartpole-balance-sparse.yml │ ├── dmc-cartpole-balance.yml │ ├── dmc-cartpole-swingup-sparse.yml │ ├── dmc-cartpole-swingup.yml │ ├── dmc-cheetah-run.yml │ ├── dmc-finger-spin.yml │ ├── dmc-finger-turn-easy.yml │ ├── dmc-finger-turn-hard.yml │ ├── dmc-hopper-hop.yml │ ├── dmc-hopper-stand.yml │ ├── dmc-pendulum-swingup.yml │ ├── dmc-quadruped-run.yml │ ├── dmc-quadruped-walk.yml │ ├── dmc-reacher-easy.yml │ ├── dmc-reacher-hard.yml │ ├── dmc-walker-run.yml │ ├── dmc-walker-stand.yml │ ├── dmc-walker-walk.yml │ └── p2e-dmc-walker-walk.yml ├── envs │ ├── envs.py │ └── wrappers.py ├── modules │ ├── actor.py │ ├── critic.py │ ├── decoder.py │ ├── encoder.py │ ├── model.py │ └── one_step_model.py └── utils │ ├── buffer.py │ └── utils.py ├── main.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | venv 3 | runs 4 | __pycache__ 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/README.md -------------------------------------------------------------------------------- /dreamer/algorithms/dreamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/algorithms/dreamer.py -------------------------------------------------------------------------------- /dreamer/algorithms/plan2explore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/algorithms/plan2explore.py -------------------------------------------------------------------------------- /dreamer/configs/atari-boxing-v4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/atari-boxing-v4.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-acrobot-swingup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-acrobot-swingup.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-ball-in-cup-catch.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-ball-in-cup-catch.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-cartpole-balance-sparse.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-cartpole-balance-sparse.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-cartpole-balance.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-cartpole-balance.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-cartpole-swingup-sparse.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-cartpole-swingup-sparse.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-cartpole-swingup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-cartpole-swingup.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-cheetah-run.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-cheetah-run.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-finger-spin.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-finger-spin.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-finger-turn-easy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-finger-turn-easy.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-finger-turn-hard.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-finger-turn-hard.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-hopper-hop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-hopper-hop.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-hopper-stand.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-hopper-stand.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-pendulum-swingup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-pendulum-swingup.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-quadruped-run.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-quadruped-run.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-quadruped-walk.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-quadruped-walk.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-reacher-easy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-reacher-easy.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-reacher-hard.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-reacher-hard.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-walker-run.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-walker-run.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-walker-stand.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-walker-stand.yml -------------------------------------------------------------------------------- /dreamer/configs/dmc-walker-walk.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/dmc-walker-walk.yml -------------------------------------------------------------------------------- /dreamer/configs/p2e-dmc-walker-walk.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/configs/p2e-dmc-walker-walk.yml -------------------------------------------------------------------------------- /dreamer/envs/envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/envs/envs.py -------------------------------------------------------------------------------- /dreamer/envs/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/envs/wrappers.py -------------------------------------------------------------------------------- /dreamer/modules/actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/actor.py -------------------------------------------------------------------------------- /dreamer/modules/critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/critic.py -------------------------------------------------------------------------------- /dreamer/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/decoder.py -------------------------------------------------------------------------------- /dreamer/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/encoder.py -------------------------------------------------------------------------------- /dreamer/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/model.py -------------------------------------------------------------------------------- /dreamer/modules/one_step_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/modules/one_step_model.py -------------------------------------------------------------------------------- /dreamer/utils/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/utils/buffer.py -------------------------------------------------------------------------------- /dreamer/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/dreamer/utils/utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kc-ml2/SimpleDreamer/HEAD/requirements.txt --------------------------------------------------------------------------------