├── LICENSE ├── README.md ├── docker ├── Dockerfile └── build.sh ├── install_sc2.sh ├── requirements.txt ├── run.sh ├── run_interactive.sh └── src ├── __init__.py ├── components ├── __init__.py ├── action_selectors.py ├── episode_buffer.py ├── epsilon_schedules.py └── transforms.py ├── config ├── algs │ └── casec.yaml ├── default.yaml └── envs │ ├── aloha.yaml │ ├── disperse.yaml │ ├── gather.yaml │ ├── hallway.yaml │ ├── pursuit.yaml │ ├── sc2.yaml │ ├── sc2_beta.yaml │ └── sensor.yaml ├── controllers ├── __init__.py ├── basic_controller.py └── casec_controller.py ├── envs ├── __init__.py ├── aloha.py ├── disperse.py ├── gather.py ├── hallway.py ├── multiagentenv.py ├── pursuit.py ├── sensors.py └── starcraft2 │ ├── __init__.py │ ├── maps │ ├── __init__.py │ └── smac_maps.py │ └── starcraft2.py ├── learners ├── __init__.py ├── casec_learner.py ├── coma_learner.py ├── q_learner.py └── qtran_learner.py ├── main.py ├── modules ├── __init__.py ├── action_encoders │ ├── __init__.py │ └── obs_reward_encoder.py ├── agents │ ├── __init__.py │ └── rnn_agent.py ├── critics │ ├── __init__.py │ └── coma.py └── mixers │ ├── __init__.py │ ├── qmix.py │ ├── qtran.py │ └── vdn.py ├── run.py ├── runners ├── __init__.py ├── episode_runner.py └── parallel_runner.py └── utils ├── dict2namedtuple.py ├── logging.py ├── rl_utils.py └── timehelper.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/docker/build.sh -------------------------------------------------------------------------------- /install_sc2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/install_sc2.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/run.sh -------------------------------------------------------------------------------- /run_interactive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/run_interactive.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/components/action_selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/components/action_selectors.py -------------------------------------------------------------------------------- /src/components/episode_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/components/episode_buffer.py -------------------------------------------------------------------------------- /src/components/epsilon_schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/components/epsilon_schedules.py -------------------------------------------------------------------------------- /src/components/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/components/transforms.py -------------------------------------------------------------------------------- /src/config/algs/casec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/algs/casec.yaml -------------------------------------------------------------------------------- /src/config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/default.yaml -------------------------------------------------------------------------------- /src/config/envs/aloha.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/aloha.yaml -------------------------------------------------------------------------------- /src/config/envs/disperse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/disperse.yaml -------------------------------------------------------------------------------- /src/config/envs/gather.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/gather.yaml -------------------------------------------------------------------------------- /src/config/envs/hallway.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/hallway.yaml -------------------------------------------------------------------------------- /src/config/envs/pursuit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/pursuit.yaml -------------------------------------------------------------------------------- /src/config/envs/sc2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/sc2.yaml -------------------------------------------------------------------------------- /src/config/envs/sc2_beta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/sc2_beta.yaml -------------------------------------------------------------------------------- /src/config/envs/sensor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/config/envs/sensor.yaml -------------------------------------------------------------------------------- /src/controllers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/controllers/__init__.py -------------------------------------------------------------------------------- /src/controllers/basic_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/controllers/basic_controller.py -------------------------------------------------------------------------------- /src/controllers/casec_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/controllers/casec_controller.py -------------------------------------------------------------------------------- /src/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/__init__.py -------------------------------------------------------------------------------- /src/envs/aloha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/aloha.py -------------------------------------------------------------------------------- /src/envs/disperse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/disperse.py -------------------------------------------------------------------------------- /src/envs/gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/gather.py -------------------------------------------------------------------------------- /src/envs/hallway.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/hallway.py -------------------------------------------------------------------------------- /src/envs/multiagentenv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/multiagentenv.py -------------------------------------------------------------------------------- /src/envs/pursuit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/pursuit.py -------------------------------------------------------------------------------- /src/envs/sensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/sensors.py -------------------------------------------------------------------------------- /src/envs/starcraft2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/starcraft2/__init__.py -------------------------------------------------------------------------------- /src/envs/starcraft2/maps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/starcraft2/maps/__init__.py -------------------------------------------------------------------------------- /src/envs/starcraft2/maps/smac_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/starcraft2/maps/smac_maps.py -------------------------------------------------------------------------------- /src/envs/starcraft2/starcraft2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/envs/starcraft2/starcraft2.py -------------------------------------------------------------------------------- /src/learners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/learners/__init__.py -------------------------------------------------------------------------------- /src/learners/casec_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/learners/casec_learner.py -------------------------------------------------------------------------------- /src/learners/coma_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/learners/coma_learner.py -------------------------------------------------------------------------------- /src/learners/q_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/learners/q_learner.py -------------------------------------------------------------------------------- /src/learners/qtran_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/learners/qtran_learner.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/main.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/action_encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/action_encoders/__init__.py -------------------------------------------------------------------------------- /src/modules/action_encoders/obs_reward_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/action_encoders/obs_reward_encoder.py -------------------------------------------------------------------------------- /src/modules/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/agents/__init__.py -------------------------------------------------------------------------------- /src/modules/agents/rnn_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/agents/rnn_agent.py -------------------------------------------------------------------------------- /src/modules/critics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/critics/coma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/critics/coma.py -------------------------------------------------------------------------------- /src/modules/mixers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/mixers/qmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/mixers/qmix.py -------------------------------------------------------------------------------- /src/modules/mixers/qtran.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/mixers/qtran.py -------------------------------------------------------------------------------- /src/modules/mixers/vdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/modules/mixers/vdn.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/run.py -------------------------------------------------------------------------------- /src/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/runners/__init__.py -------------------------------------------------------------------------------- /src/runners/episode_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/runners/episode_runner.py -------------------------------------------------------------------------------- /src/runners/parallel_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/runners/parallel_runner.py -------------------------------------------------------------------------------- /src/utils/dict2namedtuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/utils/dict2namedtuple.py -------------------------------------------------------------------------------- /src/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/utils/logging.py -------------------------------------------------------------------------------- /src/utils/rl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/utils/rl_utils.py -------------------------------------------------------------------------------- /src/utils/timehelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/CASEC-MACO-benchmark/HEAD/src/utils/timehelper.py --------------------------------------------------------------------------------