├── AUTHORS ├── LICENSE ├── README.md ├── dmc_zeroshot_plan2explore.json ├── plan2explore ├── __init__.py ├── control │ ├── __init__.py │ ├── batch_env.py │ ├── dummy_env.py │ ├── mpc_agent.py │ ├── planning.py │ ├── random_episodes.py │ ├── simulate.py │ ├── temporal_difference.py │ └── wrappers.py ├── distributions │ ├── __init__.py │ ├── one_hot.py │ └── tanh_normal.py ├── models │ ├── __init__.py │ ├── base.py │ └── rssm.py ├── networks │ ├── __init__.py │ ├── basic.py │ ├── conv.py │ ├── one_step_model.py │ └── proprio.py ├── scripts │ ├── __init__.py │ ├── configs.py │ ├── fileutil_fetch.py │ ├── objectives.py │ ├── sync.py │ ├── tasks.py │ └── train.py ├── tools │ ├── __init__.py │ ├── attr_dict.py │ ├── bind.py │ ├── chunk_sequence.py │ ├── copy_weights.py │ ├── count_dataset.py │ ├── count_weights.py │ ├── custom_optimizer.py │ ├── filter_variables_lib.py │ ├── gif_summary.py │ ├── image_strip_summary.py │ ├── mask.py │ ├── metrics.py │ ├── nested.py │ ├── numpy_episodes.py │ ├── overshooting.py │ ├── preprocess.py │ ├── reshape_as.py │ ├── schedule.py │ ├── shape.py │ ├── streaming_mean.py │ ├── summary.py │ ├── target_network.py │ └── unroll.py └── training │ ├── __init__.py │ ├── define_model.py │ ├── define_summaries.py │ ├── running.py │ ├── trainer.py │ └── utility.py └── requirements.txt /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/AUTHORS -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/README.md -------------------------------------------------------------------------------- /dmc_zeroshot_plan2explore.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/dmc_zeroshot_plan2explore.json -------------------------------------------------------------------------------- /plan2explore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/__init__.py -------------------------------------------------------------------------------- /plan2explore/control/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/__init__.py -------------------------------------------------------------------------------- /plan2explore/control/batch_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/batch_env.py -------------------------------------------------------------------------------- /plan2explore/control/dummy_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/dummy_env.py -------------------------------------------------------------------------------- /plan2explore/control/mpc_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/mpc_agent.py -------------------------------------------------------------------------------- /plan2explore/control/planning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/planning.py -------------------------------------------------------------------------------- /plan2explore/control/random_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/random_episodes.py -------------------------------------------------------------------------------- /plan2explore/control/simulate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/simulate.py -------------------------------------------------------------------------------- /plan2explore/control/temporal_difference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/temporal_difference.py -------------------------------------------------------------------------------- /plan2explore/control/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/control/wrappers.py -------------------------------------------------------------------------------- /plan2explore/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/distributions/__init__.py -------------------------------------------------------------------------------- /plan2explore/distributions/one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/distributions/one_hot.py -------------------------------------------------------------------------------- /plan2explore/distributions/tanh_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/distributions/tanh_normal.py -------------------------------------------------------------------------------- /plan2explore/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/models/__init__.py -------------------------------------------------------------------------------- /plan2explore/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/models/base.py -------------------------------------------------------------------------------- /plan2explore/models/rssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/models/rssm.py -------------------------------------------------------------------------------- /plan2explore/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/networks/__init__.py -------------------------------------------------------------------------------- /plan2explore/networks/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/networks/basic.py -------------------------------------------------------------------------------- /plan2explore/networks/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/networks/conv.py -------------------------------------------------------------------------------- /plan2explore/networks/one_step_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/networks/one_step_model.py -------------------------------------------------------------------------------- /plan2explore/networks/proprio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/networks/proprio.py -------------------------------------------------------------------------------- /plan2explore/scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/__init__.py -------------------------------------------------------------------------------- /plan2explore/scripts/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/configs.py -------------------------------------------------------------------------------- /plan2explore/scripts/fileutil_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/fileutil_fetch.py -------------------------------------------------------------------------------- /plan2explore/scripts/objectives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/objectives.py -------------------------------------------------------------------------------- /plan2explore/scripts/sync.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/sync.py -------------------------------------------------------------------------------- /plan2explore/scripts/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/tasks.py -------------------------------------------------------------------------------- /plan2explore/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/scripts/train.py -------------------------------------------------------------------------------- /plan2explore/tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/__init__.py -------------------------------------------------------------------------------- /plan2explore/tools/attr_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/attr_dict.py -------------------------------------------------------------------------------- /plan2explore/tools/bind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/bind.py -------------------------------------------------------------------------------- /plan2explore/tools/chunk_sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/chunk_sequence.py -------------------------------------------------------------------------------- /plan2explore/tools/copy_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/copy_weights.py -------------------------------------------------------------------------------- /plan2explore/tools/count_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/count_dataset.py -------------------------------------------------------------------------------- /plan2explore/tools/count_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/count_weights.py -------------------------------------------------------------------------------- /plan2explore/tools/custom_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/custom_optimizer.py -------------------------------------------------------------------------------- /plan2explore/tools/filter_variables_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/filter_variables_lib.py -------------------------------------------------------------------------------- /plan2explore/tools/gif_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/gif_summary.py -------------------------------------------------------------------------------- /plan2explore/tools/image_strip_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/image_strip_summary.py -------------------------------------------------------------------------------- /plan2explore/tools/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/mask.py -------------------------------------------------------------------------------- /plan2explore/tools/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/metrics.py -------------------------------------------------------------------------------- /plan2explore/tools/nested.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/nested.py -------------------------------------------------------------------------------- /plan2explore/tools/numpy_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/numpy_episodes.py -------------------------------------------------------------------------------- /plan2explore/tools/overshooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/overshooting.py -------------------------------------------------------------------------------- /plan2explore/tools/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/preprocess.py -------------------------------------------------------------------------------- /plan2explore/tools/reshape_as.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/reshape_as.py -------------------------------------------------------------------------------- /plan2explore/tools/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/schedule.py -------------------------------------------------------------------------------- /plan2explore/tools/shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/shape.py -------------------------------------------------------------------------------- /plan2explore/tools/streaming_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/streaming_mean.py -------------------------------------------------------------------------------- /plan2explore/tools/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/summary.py -------------------------------------------------------------------------------- /plan2explore/tools/target_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/target_network.py -------------------------------------------------------------------------------- /plan2explore/tools/unroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/tools/unroll.py -------------------------------------------------------------------------------- /plan2explore/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/__init__.py -------------------------------------------------------------------------------- /plan2explore/training/define_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/define_model.py -------------------------------------------------------------------------------- /plan2explore/training/define_summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/define_summaries.py -------------------------------------------------------------------------------- /plan2explore/training/running.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/running.py -------------------------------------------------------------------------------- /plan2explore/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/trainer.py -------------------------------------------------------------------------------- /plan2explore/training/utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/plan2explore/training/utility.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramanans1/plan2explore/HEAD/requirements.txt --------------------------------------------------------------------------------