├── .DS_Store ├── .cloudbase └── container │ └── debug.json ├── .idea ├── .gitignore ├── My_PlaNet.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── ELBO1.jpg ├── ELBO2.png ├── ELBO_proof1.png ├── ELBO_proof2.png ├── RSSM.png ├── __pycache__ ├── agent.cpython-310.pyc ├── agent.cpython-38.pyc ├── model.cpython-310.pyc ├── model.cpython-38.pyc ├── train.cpython-310.pyc ├── utils.cpython-310.pyc ├── utils.cpython-38.pyc ├── viewer.cpython-310.pyc ├── viewer.cpython-38.pyc ├── wrappers.cpython-310.pyc └── wrappers.cpython-38.pyc ├── agent.py ├── cheetah run.gif ├── log ├── cheetah_run │ ├── 20250309_0057 │ │ ├── args.json │ │ └── events.out.tfevents.1741453025.ubuntu22.14880.0 │ ├── 20250309_1119 │ │ ├── args.json │ │ ├── encoder.pth │ │ ├── events.out.tfevents.1741490342.ubuntu22.5565.0 │ │ ├── obs_model.pth │ │ ├── reward_model.pth │ │ └── rssm.pth │ ├── 20250309_1216 │ │ ├── args.json │ │ ├── encoder.pth │ │ ├── events.out.tfevents.1741493819.ubuntu22.11670.0 │ │ ├── obs_model.pth │ │ ├── reward_model.pth │ │ └── rssm.pth │ └── 20250313_1636 │ │ ├── args.json │ │ └── events.out.tfevents.1741854981.MacBook-Air-38.local.32225.0 ├── reacher_easy │ ├── 20250309_0131 │ │ ├── args.json │ │ └── events.out.tfevents.1741455069.ubuntu22.27936.0 │ └── 20250309_1825 │ │ ├── args.json │ │ ├── encoder.pth │ │ ├── events.out.tfevents.1741515901.ubuntu22.74135.0 │ │ ├── obs_model.pth │ │ ├── reward_model.pth │ │ └── rssm.pth └── walker_run │ └── 20250309_0102 │ ├── args.json │ └── events.out.tfevents.1741453344.ubuntu22.16634.0 ├── model.py ├── reacher easy.gif ├── readme.md ├── requirements.txt ├── test.py ├── train.py ├── utils.py ├── video_prediction.py ├── viewer.py └── wrappers.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.DS_Store -------------------------------------------------------------------------------- /.cloudbase/container/debug.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.cloudbase/container/debug.json -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/My_PlaNet.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/My_PlaNet.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /ELBO1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/ELBO1.jpg -------------------------------------------------------------------------------- /ELBO2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/ELBO2.png -------------------------------------------------------------------------------- /ELBO_proof1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/ELBO_proof1.png -------------------------------------------------------------------------------- /ELBO_proof2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/ELBO_proof2.png -------------------------------------------------------------------------------- /RSSM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/RSSM.png -------------------------------------------------------------------------------- /__pycache__/agent.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/agent.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/agent.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/agent.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/model.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/model.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/train.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/train.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/utils.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/viewer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/viewer.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/viewer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/viewer.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/wrappers.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/wrappers.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/wrappers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/__pycache__/wrappers.cpython-38.pyc -------------------------------------------------------------------------------- /agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/agent.py -------------------------------------------------------------------------------- /cheetah run.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/cheetah run.gif -------------------------------------------------------------------------------- /log/cheetah_run/20250309_0057/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_0057/args.json -------------------------------------------------------------------------------- /log/cheetah_run/20250309_0057/events.out.tfevents.1741453025.ubuntu22.14880.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_0057/events.out.tfevents.1741453025.ubuntu22.14880.0 -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/args.json -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/encoder.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/encoder.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/events.out.tfevents.1741490342.ubuntu22.5565.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/events.out.tfevents.1741490342.ubuntu22.5565.0 -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/obs_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/obs_model.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/reward_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/reward_model.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1119/rssm.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1119/rssm.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/args.json -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/encoder.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/encoder.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/events.out.tfevents.1741493819.ubuntu22.11670.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/events.out.tfevents.1741493819.ubuntu22.11670.0 -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/obs_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/obs_model.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/reward_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/reward_model.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250309_1216/rssm.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250309_1216/rssm.pth -------------------------------------------------------------------------------- /log/cheetah_run/20250313_1636/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250313_1636/args.json -------------------------------------------------------------------------------- /log/cheetah_run/20250313_1636/events.out.tfevents.1741854981.MacBook-Air-38.local.32225.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/cheetah_run/20250313_1636/events.out.tfevents.1741854981.MacBook-Air-38.local.32225.0 -------------------------------------------------------------------------------- /log/reacher_easy/20250309_0131/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_0131/args.json -------------------------------------------------------------------------------- /log/reacher_easy/20250309_0131/events.out.tfevents.1741455069.ubuntu22.27936.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_0131/events.out.tfevents.1741455069.ubuntu22.27936.0 -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/args.json -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/encoder.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/encoder.pth -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/events.out.tfevents.1741515901.ubuntu22.74135.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/events.out.tfevents.1741515901.ubuntu22.74135.0 -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/obs_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/obs_model.pth -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/reward_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/reward_model.pth -------------------------------------------------------------------------------- /log/reacher_easy/20250309_1825/rssm.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/reacher_easy/20250309_1825/rssm.pth -------------------------------------------------------------------------------- /log/walker_run/20250309_0102/args.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/walker_run/20250309_0102/args.json -------------------------------------------------------------------------------- /log/walker_run/20250309_0102/events.out.tfevents.1741453344.ubuntu22.16634.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/log/walker_run/20250309_0102/events.out.tfevents.1741453344.ubuntu22.16634.0 -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/model.py -------------------------------------------------------------------------------- /reacher easy.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/reacher easy.gif -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/utils.py -------------------------------------------------------------------------------- /video_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/video_prediction.py -------------------------------------------------------------------------------- /viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/viewer.py -------------------------------------------------------------------------------- /wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jinhsi555/My_PlaNet/HEAD/wrappers.py --------------------------------------------------------------------------------