├── .gitignore ├── LICENSE ├── README.md ├── buffer.py ├── configs ├── cartpole.yaml ├── minigrid.yaml ├── mortar_mayhem_grid.yaml ├── mystery_path_grid.yaml └── poc_memory_env.yaml ├── docs └── assets │ ├── mmg_10.png │ ├── mpg_off.png │ └── trxl_architecture.png ├── enjoy.py ├── environments ├── __init__.py ├── cartpole_env.py ├── memory_gym_env.py ├── minigrid_env.py └── poc_memory_env.py ├── model.py ├── models ├── minigrid.nn ├── mortar_mayhem_grid_trxl.nn └── mystery_path_grid.nn ├── requirements.txt ├── train.py ├── trainer.py ├── transformer.py ├── utils.py ├── worker.py └── yaml_parser.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | .vscode/ 3 | summaries/ 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/README.md -------------------------------------------------------------------------------- /buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/buffer.py -------------------------------------------------------------------------------- /configs/cartpole.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/configs/cartpole.yaml -------------------------------------------------------------------------------- /configs/minigrid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/configs/minigrid.yaml -------------------------------------------------------------------------------- /configs/mortar_mayhem_grid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/configs/mortar_mayhem_grid.yaml -------------------------------------------------------------------------------- /configs/mystery_path_grid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/configs/mystery_path_grid.yaml -------------------------------------------------------------------------------- /configs/poc_memory_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/configs/poc_memory_env.yaml -------------------------------------------------------------------------------- /docs/assets/mmg_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/docs/assets/mmg_10.png -------------------------------------------------------------------------------- /docs/assets/mpg_off.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/docs/assets/mpg_off.png -------------------------------------------------------------------------------- /docs/assets/trxl_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/docs/assets/trxl_architecture.png -------------------------------------------------------------------------------- /enjoy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/enjoy.py -------------------------------------------------------------------------------- /environments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /environments/cartpole_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/environments/cartpole_env.py -------------------------------------------------------------------------------- /environments/memory_gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/environments/memory_gym_env.py -------------------------------------------------------------------------------- /environments/minigrid_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/environments/minigrid_env.py -------------------------------------------------------------------------------- /environments/poc_memory_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/environments/poc_memory_env.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/model.py -------------------------------------------------------------------------------- /models/minigrid.nn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/models/minigrid.nn -------------------------------------------------------------------------------- /models/mortar_mayhem_grid_trxl.nn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/models/mortar_mayhem_grid_trxl.nn -------------------------------------------------------------------------------- /models/mystery_path_grid.nn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/models/mystery_path_grid.nn -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/trainer.py -------------------------------------------------------------------------------- /transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/transformer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/utils.py -------------------------------------------------------------------------------- /worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/worker.py -------------------------------------------------------------------------------- /yaml_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MarcoMeter/episodic-transformer-memory-ppo/HEAD/yaml_parser.py --------------------------------------------------------------------------------