├── LICENSE ├── README.md ├── adroit.sh ├── antmaze.sh ├── config └── vqvae.py ├── environment.yml ├── img └── trainandtest.png ├── latentplan ├── __init__.py ├── datasets │ ├── __init__.py │ ├── d4rl.py │ ├── preprocessing.py │ └── sequence.py ├── models │ ├── __init__.py │ ├── autoencoders.py │ ├── ein.py │ ├── embeddings.py │ ├── transformers.py │ └── vqvae.py ├── search │ ├── __init__.py │ ├── core.py │ ├── optimizer.py │ ├── sampling.py │ └── utils.py └── utils │ ├── __init__.py │ ├── arrays.py │ ├── config.py │ ├── discretization.py │ ├── git_utils.py │ ├── progress.py │ ├── rendering.py │ ├── serialization.py │ ├── setup.py │ ├── timer.py │ ├── training.py │ └── video.py ├── loco.sh ├── plotting ├── adtable.py ├── assets │ ├── ant.xml │ ├── half_cheetah.xml │ ├── hopper.xml │ └── walker2d.xml ├── dynamics.py ├── fullobsenv.py ├── horizon_scale.py ├── latentstep.py ├── parselog.py ├── plot.py ├── read_results.py ├── sampling.py ├── scores.py ├── table.py └── threshold.py ├── scripts ├── explorelatent.py ├── mazegen.py ├── plan.py ├── train.py ├── trainprior.py └── visualizedata.py └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/README.md -------------------------------------------------------------------------------- /adroit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/adroit.sh -------------------------------------------------------------------------------- /antmaze.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/antmaze.sh -------------------------------------------------------------------------------- /config/vqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/config/vqvae.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/environment.yml -------------------------------------------------------------------------------- /img/trainandtest.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/img/trainandtest.png -------------------------------------------------------------------------------- /latentplan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /latentplan/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/datasets/__init__.py -------------------------------------------------------------------------------- /latentplan/datasets/d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/datasets/d4rl.py -------------------------------------------------------------------------------- /latentplan/datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/datasets/preprocessing.py -------------------------------------------------------------------------------- /latentplan/datasets/sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/datasets/sequence.py -------------------------------------------------------------------------------- /latentplan/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /latentplan/models/autoencoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/models/autoencoders.py -------------------------------------------------------------------------------- /latentplan/models/ein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/models/ein.py -------------------------------------------------------------------------------- /latentplan/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/models/embeddings.py -------------------------------------------------------------------------------- /latentplan/models/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/models/transformers.py -------------------------------------------------------------------------------- /latentplan/models/vqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/models/vqvae.py -------------------------------------------------------------------------------- /latentplan/search/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/search/__init__.py -------------------------------------------------------------------------------- /latentplan/search/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/search/core.py -------------------------------------------------------------------------------- /latentplan/search/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/search/optimizer.py -------------------------------------------------------------------------------- /latentplan/search/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/search/sampling.py -------------------------------------------------------------------------------- /latentplan/search/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/search/utils.py -------------------------------------------------------------------------------- /latentplan/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/__init__.py -------------------------------------------------------------------------------- /latentplan/utils/arrays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/arrays.py -------------------------------------------------------------------------------- /latentplan/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/config.py -------------------------------------------------------------------------------- /latentplan/utils/discretization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/discretization.py -------------------------------------------------------------------------------- /latentplan/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/git_utils.py -------------------------------------------------------------------------------- /latentplan/utils/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/progress.py -------------------------------------------------------------------------------- /latentplan/utils/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/rendering.py -------------------------------------------------------------------------------- /latentplan/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/serialization.py -------------------------------------------------------------------------------- /latentplan/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/setup.py -------------------------------------------------------------------------------- /latentplan/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/timer.py -------------------------------------------------------------------------------- /latentplan/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/training.py -------------------------------------------------------------------------------- /latentplan/utils/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/latentplan/utils/video.py -------------------------------------------------------------------------------- /loco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/loco.sh -------------------------------------------------------------------------------- /plotting/adtable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/adtable.py -------------------------------------------------------------------------------- /plotting/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/assets/ant.xml -------------------------------------------------------------------------------- /plotting/assets/half_cheetah.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/assets/half_cheetah.xml -------------------------------------------------------------------------------- /plotting/assets/hopper.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/assets/hopper.xml -------------------------------------------------------------------------------- /plotting/assets/walker2d.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/assets/walker2d.xml -------------------------------------------------------------------------------- /plotting/dynamics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/dynamics.py -------------------------------------------------------------------------------- /plotting/fullobsenv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/fullobsenv.py -------------------------------------------------------------------------------- /plotting/horizon_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/horizon_scale.py -------------------------------------------------------------------------------- /plotting/latentstep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/latentstep.py -------------------------------------------------------------------------------- /plotting/parselog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/parselog.py -------------------------------------------------------------------------------- /plotting/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/plot.py -------------------------------------------------------------------------------- /plotting/read_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/read_results.py -------------------------------------------------------------------------------- /plotting/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/sampling.py -------------------------------------------------------------------------------- /plotting/scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/scores.py -------------------------------------------------------------------------------- /plotting/table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/table.py -------------------------------------------------------------------------------- /plotting/threshold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/plotting/threshold.py -------------------------------------------------------------------------------- /scripts/explorelatent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/explorelatent.py -------------------------------------------------------------------------------- /scripts/mazegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/mazegen.py -------------------------------------------------------------------------------- /scripts/plan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/plan.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/trainprior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/trainprior.py -------------------------------------------------------------------------------- /scripts/visualizedata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/scripts/visualizedata.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhengyaoJiang/latentplan/HEAD/setup.py --------------------------------------------------------------------------------