├── .gitignore ├── README.md ├── __init__.py ├── main_mbpo.py ├── make_plot.py ├── model.py ├── predict_env.py ├── results ├── hopper.png └── walker2d.png ├── sac ├── LICENSE ├── README.md ├── __init__.py ├── main.py ├── model.py ├── replay_memory.py ├── sac.py └── utils.py ├── sample_env.py └── tf_models ├── __init__.py ├── bnn.py ├── constructor.py ├── fake_env.py ├── fc.py ├── tf_logging.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | data 2 | .idea 3 | *__pycache__* 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /main_mbpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/main_mbpo.py -------------------------------------------------------------------------------- /make_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/make_plot.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/model.py -------------------------------------------------------------------------------- /predict_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/predict_env.py -------------------------------------------------------------------------------- /results/hopper.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/results/hopper.png -------------------------------------------------------------------------------- /results/walker2d.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/results/walker2d.png -------------------------------------------------------------------------------- /sac/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/LICENSE -------------------------------------------------------------------------------- /sac/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/README.md -------------------------------------------------------------------------------- /sac/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sac/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/main.py -------------------------------------------------------------------------------- /sac/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/model.py -------------------------------------------------------------------------------- /sac/replay_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/replay_memory.py -------------------------------------------------------------------------------- /sac/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/sac.py -------------------------------------------------------------------------------- /sac/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sac/utils.py -------------------------------------------------------------------------------- /sample_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/sample_env.py -------------------------------------------------------------------------------- /tf_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tf_models/bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/bnn.py -------------------------------------------------------------------------------- /tf_models/constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/constructor.py -------------------------------------------------------------------------------- /tf_models/fake_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/fake_env.py -------------------------------------------------------------------------------- /tf_models/fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/fc.py -------------------------------------------------------------------------------- /tf_models/tf_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/tf_logging.py -------------------------------------------------------------------------------- /tf_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xingyu-Lin/mbpo_pytorch/HEAD/tf_models/utils.py --------------------------------------------------------------------------------