├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── feature_request.md │ └── question.md ├── dependabot.yml └── workflows │ └── tests.yaml ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── dreamer ├── __init__.py ├── agents │ ├── __init__.py │ ├── atari_dreamer_agent.py │ ├── dmc_dreamer_agent.py │ └── dreamer_agent.py ├── algos │ ├── __init__.py │ ├── dreamer_algo.py │ └── replay.py ├── envs │ ├── __init__.py │ ├── action_repeat.py │ ├── atari.py │ ├── dmc.py │ ├── env.py │ ├── modified_atari.py │ ├── normalize_actions.py │ ├── one_hot.py │ ├── time_limit.py │ └── wrapper.py ├── experiments │ ├── __init__.py │ ├── configs │ │ ├── __init__.py │ │ └── atari │ │ │ ├── __init__.py │ │ │ └── atari_dreamer.py │ └── scripts │ │ └── atari │ │ ├── launch │ │ ├── launch_atari_dreamer_cpu.py │ │ └── launch_atari_dreamer_gpu.py │ │ └── train │ │ ├── atari_dreamer_cpu.py │ │ └── atari_dreamer_gpu.py ├── models │ ├── __init__.py │ ├── action.py │ ├── agent.py │ ├── dense.py │ ├── distribution.py │ ├── observation.py │ └── rnns.py └── utils │ ├── __init__.py │ ├── launching │ ├── __init__.py │ └── exp_launcher.py │ ├── logging.py │ └── module.py ├── main.py ├── main_dmc.py ├── poetry.lock ├── pyproject.toml ├── requirements.txt ├── setup ├── linux_cu118 │ ├── poetry.lock │ ├── pyproject.toml │ └── requirements.txt └── windows_cu118 │ ├── poetry.lock │ ├── pyproject.toml │ └── requirements.txt └── tests ├── __init__.py ├── test_action.py ├── test_agent.py ├── test_dense.py ├── test_distribution.py ├── test_main.py ├── test_observation.py ├── test_rnns.py └── test_utils.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.github/ISSUE_TEMPLATE/question.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.github/workflows/tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /dreamer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/agents/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/agents/atari_dreamer_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/agents/atari_dreamer_agent.py -------------------------------------------------------------------------------- /dreamer/agents/dmc_dreamer_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/agents/dmc_dreamer_agent.py -------------------------------------------------------------------------------- /dreamer/agents/dreamer_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/agents/dreamer_agent.py -------------------------------------------------------------------------------- /dreamer/algos/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/algos/dreamer_algo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/algos/dreamer_algo.py -------------------------------------------------------------------------------- /dreamer/algos/replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/algos/replay.py -------------------------------------------------------------------------------- /dreamer/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/envs/action_repeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/action_repeat.py -------------------------------------------------------------------------------- /dreamer/envs/atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/atari.py -------------------------------------------------------------------------------- /dreamer/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/dmc.py -------------------------------------------------------------------------------- /dreamer/envs/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/env.py -------------------------------------------------------------------------------- /dreamer/envs/modified_atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/modified_atari.py -------------------------------------------------------------------------------- /dreamer/envs/normalize_actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/normalize_actions.py -------------------------------------------------------------------------------- /dreamer/envs/one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/one_hot.py -------------------------------------------------------------------------------- /dreamer/envs/time_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/time_limit.py -------------------------------------------------------------------------------- /dreamer/envs/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/envs/wrapper.py -------------------------------------------------------------------------------- /dreamer/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/experiments/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/experiments/configs/atari/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/experiments/configs/atari/atari_dreamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/experiments/configs/atari/atari_dreamer.py -------------------------------------------------------------------------------- /dreamer/experiments/scripts/atari/launch/launch_atari_dreamer_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/experiments/scripts/atari/launch/launch_atari_dreamer_cpu.py -------------------------------------------------------------------------------- /dreamer/experiments/scripts/atari/launch/launch_atari_dreamer_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/experiments/scripts/atari/launch/launch_atari_dreamer_gpu.py -------------------------------------------------------------------------------- /dreamer/experiments/scripts/atari/train/atari_dreamer_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/experiments/scripts/atari/train/atari_dreamer_cpu.py -------------------------------------------------------------------------------- /dreamer/experiments/scripts/atari/train/atari_dreamer_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/experiments/scripts/atari/train/atari_dreamer_gpu.py -------------------------------------------------------------------------------- /dreamer/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/models/action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/action.py -------------------------------------------------------------------------------- /dreamer/models/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/agent.py -------------------------------------------------------------------------------- /dreamer/models/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/dense.py -------------------------------------------------------------------------------- /dreamer/models/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/distribution.py -------------------------------------------------------------------------------- /dreamer/models/observation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/observation.py -------------------------------------------------------------------------------- /dreamer/models/rnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/models/rnns.py -------------------------------------------------------------------------------- /dreamer/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/utils/launching/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dreamer/utils/launching/exp_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/utils/launching/exp_launcher.py -------------------------------------------------------------------------------- /dreamer/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/utils/logging.py -------------------------------------------------------------------------------- /dreamer/utils/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/dreamer/utils/module.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/main.py -------------------------------------------------------------------------------- /main_dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/main_dmc.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup/linux_cu118/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/linux_cu118/poetry.lock -------------------------------------------------------------------------------- /setup/linux_cu118/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/linux_cu118/pyproject.toml -------------------------------------------------------------------------------- /setup/linux_cu118/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/linux_cu118/requirements.txt -------------------------------------------------------------------------------- /setup/windows_cu118/poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/windows_cu118/poetry.lock -------------------------------------------------------------------------------- /setup/windows_cu118/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/windows_cu118/pyproject.toml -------------------------------------------------------------------------------- /setup/windows_cu118/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/setup/windows_cu118/requirements.txt -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_action.py -------------------------------------------------------------------------------- /tests/test_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_agent.py -------------------------------------------------------------------------------- /tests/test_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_dense.py -------------------------------------------------------------------------------- /tests/test_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_distribution.py -------------------------------------------------------------------------------- /tests/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_main.py -------------------------------------------------------------------------------- /tests/test_observation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_observation.py -------------------------------------------------------------------------------- /tests/test_rnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_rnns.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/juliusfrost/dreamer-pytorch/HEAD/tests/test_utils.py --------------------------------------------------------------------------------