├── LICENSE ├── README.md ├── create_conda_env_pql.sh ├── hydra_plugins └── pql_searchpath_plugin │ ├── __init__.py │ └── pql_searchpath_plugin.py ├── pql ├── __init__.py ├── algo │ ├── __init__.py │ ├── ac_base.py │ ├── ddpg.py │ ├── ppo.py │ ├── pql_actor.py │ ├── pql_p_learner.py │ ├── pql_v_learner.py │ └── sac.py ├── cfg │ ├── algo │ │ ├── actor_critic.yaml │ │ ├── ddpg_algo.yaml │ │ ├── ppo_algo.yaml │ │ ├── pql_algo.yaml │ │ └── sac_algo.yaml │ ├── default.yaml │ └── logging │ │ └── default.yaml ├── models │ ├── __init__.py │ └── mlp.py ├── replay │ ├── __init__.py │ ├── nstep_replay.py │ └── simple_replay.py ├── utils │ ├── __init__.py │ ├── common.py │ ├── distl_util.py │ ├── evaluator.py │ ├── isaacgym_util.py │ ├── model_util.py │ ├── noise.py │ ├── runner.py │ ├── schedule_util.py │ └── torch_util.py └── wrappers │ ├── __init__.py │ ├── flatten_ob.py │ └── reset.py ├── pql_conda_env.yml ├── pyproject.toml └── scripts ├── train_baselines.py ├── train_pql.py └── visualize.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/README.md -------------------------------------------------------------------------------- /create_conda_env_pql.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/create_conda_env_pql.sh -------------------------------------------------------------------------------- /hydra_plugins/pql_searchpath_plugin/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hydra_plugins/pql_searchpath_plugin/pql_searchpath_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/hydra_plugins/pql_searchpath_plugin/pql_searchpath_plugin.py -------------------------------------------------------------------------------- /pql/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/__init__.py -------------------------------------------------------------------------------- /pql/algo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/__init__.py -------------------------------------------------------------------------------- /pql/algo/ac_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/ac_base.py -------------------------------------------------------------------------------- /pql/algo/ddpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/ddpg.py -------------------------------------------------------------------------------- /pql/algo/ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/ppo.py -------------------------------------------------------------------------------- /pql/algo/pql_actor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/pql_actor.py -------------------------------------------------------------------------------- /pql/algo/pql_p_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/pql_p_learner.py -------------------------------------------------------------------------------- /pql/algo/pql_v_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/pql_v_learner.py -------------------------------------------------------------------------------- /pql/algo/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/algo/sac.py -------------------------------------------------------------------------------- /pql/cfg/algo/actor_critic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/algo/actor_critic.yaml -------------------------------------------------------------------------------- /pql/cfg/algo/ddpg_algo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/algo/ddpg_algo.yaml -------------------------------------------------------------------------------- /pql/cfg/algo/ppo_algo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/algo/ppo_algo.yaml -------------------------------------------------------------------------------- /pql/cfg/algo/pql_algo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/algo/pql_algo.yaml -------------------------------------------------------------------------------- /pql/cfg/algo/sac_algo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/algo/sac_algo.yaml -------------------------------------------------------------------------------- /pql/cfg/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/default.yaml -------------------------------------------------------------------------------- /pql/cfg/logging/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/cfg/logging/default.yaml -------------------------------------------------------------------------------- /pql/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/models/__init__.py -------------------------------------------------------------------------------- /pql/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/models/mlp.py -------------------------------------------------------------------------------- /pql/replay/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pql/replay/nstep_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/replay/nstep_replay.py -------------------------------------------------------------------------------- /pql/replay/simple_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/replay/simple_replay.py -------------------------------------------------------------------------------- /pql/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pql/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/common.py -------------------------------------------------------------------------------- /pql/utils/distl_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/distl_util.py -------------------------------------------------------------------------------- /pql/utils/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/evaluator.py -------------------------------------------------------------------------------- /pql/utils/isaacgym_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/isaacgym_util.py -------------------------------------------------------------------------------- /pql/utils/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/model_util.py -------------------------------------------------------------------------------- /pql/utils/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/noise.py -------------------------------------------------------------------------------- /pql/utils/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/runner.py -------------------------------------------------------------------------------- /pql/utils/schedule_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/schedule_util.py -------------------------------------------------------------------------------- /pql/utils/torch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/utils/torch_util.py -------------------------------------------------------------------------------- /pql/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pql/wrappers/flatten_ob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/wrappers/flatten_ob.py -------------------------------------------------------------------------------- /pql/wrappers/reset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql/wrappers/reset.py -------------------------------------------------------------------------------- /pql_conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pql_conda_env.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/train_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/scripts/train_baselines.py -------------------------------------------------------------------------------- /scripts/train_pql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/scripts/train_pql.py -------------------------------------------------------------------------------- /scripts/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Improbable-AI/pql/HEAD/scripts/visualize.py --------------------------------------------------------------------------------