├── .gitignore ├── LICENSE ├── README.md ├── docker ├── Dockerfile └── build.sh ├── install_sc2.sh ├── requirements.txt ├── run.sh ├── run_interactive.sh └── src ├── .gitignore ├── __init__.py ├── components ├── __init__.py ├── action_selectors.py ├── episode_buffer.py ├── epsilon_schedules.py └── transforms.py ├── config ├── algs │ ├── categorical_qmix.yaml │ ├── coma_smac.yaml │ ├── iql_smac.yaml │ ├── qmix.yaml │ ├── qmix_smac.yaml │ ├── tar_qmix.yaml │ └── vdn_smac.yaml ├── default.yaml └── envs │ ├── join1.yaml │ ├── k_coloring1.yaml │ ├── k_coloring2.yaml │ ├── k_coloring3.yaml │ ├── k_coloring4.yaml │ ├── listener1.yaml │ ├── sc2.yaml │ ├── shapes.yaml │ ├── spread1.yaml │ ├── tracker1.yaml │ └── trivial1.yaml ├── controllers ├── __init__.py ├── basic_controller.py ├── cate_broadcast_comm_controller.py ├── cate_broadcast_comm_controller_full.py ├── cate_broadcast_comm_controller_not_IB.py ├── cate_pruned_broadcast_comm_controller.py └── tar_comm_controller.py ├── envs ├── __init__.py └── multiagentenv.py ├── learners ├── __init__.py ├── categorical_q_learner.py ├── coma_learner.py └── q_learner.py ├── main.py ├── modules ├── __init__.py ├── agents │ ├── __init__.py │ └── rnn_agent.py ├── comm │ ├── __init__.py │ ├── information_bottleneck_comm.py │ ├── information_bottleneck_comm_full.py │ ├── information_bottleneck_comm_not_IB.py │ ├── information_bottleneck_pruned_comm.py │ └── tar_comm.py ├── critics │ ├── __init__.py │ └── coma.py └── mixers │ ├── __init__.py │ ├── qmix.py │ └── vdn.py ├── run.py ├── runners ├── __init__.py ├── episode_runner.py ├── episode_runner_full.py ├── parallel_runner.py └── parallel_runner_x.py ├── shapes └── shapes_3x3_single_red │ ├── test.colors │ ├── test.input.npy │ ├── test.labels.npy │ ├── test.output │ ├── test.query │ ├── test.shapes │ ├── test.sizes │ ├── train.large.colors │ ├── train.large.input.npy │ ├── train.large.labels.npy │ ├── train.large.output │ ├── train.large.query │ ├── train.large.shapes │ ├── train.large.sizes │ ├── train.med.colors │ ├── train.med.input.npy │ ├── train.med.labels.npy │ ├── train.med.output │ ├── train.med.query │ ├── train.med.shapes │ ├── train.med.sizes │ ├── train.small.colors │ ├── train.small.input.npy │ ├── train.small.labels.npy │ ├── train.small.output │ ├── train.small.query │ ├── train.small.shapes │ ├── train.small.sizes │ ├── train.tiny.colors │ ├── train.tiny.input.npy │ ├── train.tiny.labels.npy │ ├── train.tiny.output │ ├── train.tiny.query │ ├── train.tiny.shapes │ ├── train.tiny.sizes │ ├── val.colors │ ├── val.input.npy │ ├── val.labels.npy │ ├── val.output │ ├── val.query │ ├── val.shapes │ └── val.sizes ├── smac_plus ├── __init__.py ├── join1.py ├── sc2_maps │ ├── 1o_10b_vs_1r.SC2Map │ ├── 1o_2r_vs_4r.SC2Map │ ├── 3s_vs_5z.SC2Map │ ├── 5z_vs_1ul.SC2Map │ ├── MMM.SC2Map │ └── bane_vs_hM.SC2Map ├── starcraft2 │ ├── __init__.py │ ├── maps │ │ └── __init__.py │ └── starcraft2.py └── tracker1.py └── utils ├── __init__.py ├── dict2namedtuple.py ├── draw_message_distributions.py ├── logging.py ├── rl_utils.py └── timehelper.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/README.md -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/docker/build.sh -------------------------------------------------------------------------------- /install_sc2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/install_sc2.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/run.sh -------------------------------------------------------------------------------- /run_interactive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/run_interactive.sh -------------------------------------------------------------------------------- /src/.gitignore: -------------------------------------------------------------------------------- 1 | tb_logs/ 2 | results/ 3 | -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/components/action_selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/components/action_selectors.py -------------------------------------------------------------------------------- /src/components/episode_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/components/episode_buffer.py -------------------------------------------------------------------------------- /src/components/epsilon_schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/components/epsilon_schedules.py -------------------------------------------------------------------------------- /src/components/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/components/transforms.py -------------------------------------------------------------------------------- /src/config/algs/categorical_qmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/categorical_qmix.yaml -------------------------------------------------------------------------------- /src/config/algs/coma_smac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/coma_smac.yaml -------------------------------------------------------------------------------- /src/config/algs/iql_smac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/iql_smac.yaml -------------------------------------------------------------------------------- /src/config/algs/qmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/qmix.yaml -------------------------------------------------------------------------------- /src/config/algs/qmix_smac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/qmix_smac.yaml -------------------------------------------------------------------------------- /src/config/algs/tar_qmix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/tar_qmix.yaml -------------------------------------------------------------------------------- /src/config/algs/vdn_smac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/algs/vdn_smac.yaml -------------------------------------------------------------------------------- /src/config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/default.yaml -------------------------------------------------------------------------------- /src/config/envs/join1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/join1.yaml -------------------------------------------------------------------------------- /src/config/envs/k_coloring1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/k_coloring1.yaml -------------------------------------------------------------------------------- /src/config/envs/k_coloring2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/k_coloring2.yaml -------------------------------------------------------------------------------- /src/config/envs/k_coloring3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/k_coloring3.yaml -------------------------------------------------------------------------------- /src/config/envs/k_coloring4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/k_coloring4.yaml -------------------------------------------------------------------------------- /src/config/envs/listener1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/listener1.yaml -------------------------------------------------------------------------------- /src/config/envs/sc2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/sc2.yaml -------------------------------------------------------------------------------- /src/config/envs/shapes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/shapes.yaml -------------------------------------------------------------------------------- /src/config/envs/spread1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/spread1.yaml -------------------------------------------------------------------------------- /src/config/envs/tracker1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/tracker1.yaml -------------------------------------------------------------------------------- /src/config/envs/trivial1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/config/envs/trivial1.yaml -------------------------------------------------------------------------------- /src/controllers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/__init__.py -------------------------------------------------------------------------------- /src/controllers/basic_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/basic_controller.py -------------------------------------------------------------------------------- /src/controllers/cate_broadcast_comm_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/cate_broadcast_comm_controller.py -------------------------------------------------------------------------------- /src/controllers/cate_broadcast_comm_controller_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/cate_broadcast_comm_controller_full.py -------------------------------------------------------------------------------- /src/controllers/cate_broadcast_comm_controller_not_IB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/cate_broadcast_comm_controller_not_IB.py -------------------------------------------------------------------------------- /src/controllers/cate_pruned_broadcast_comm_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/cate_pruned_broadcast_comm_controller.py -------------------------------------------------------------------------------- /src/controllers/tar_comm_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/controllers/tar_comm_controller.py -------------------------------------------------------------------------------- /src/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/envs/__init__.py -------------------------------------------------------------------------------- /src/envs/multiagentenv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/envs/multiagentenv.py -------------------------------------------------------------------------------- /src/learners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/learners/__init__.py -------------------------------------------------------------------------------- /src/learners/categorical_q_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/learners/categorical_q_learner.py -------------------------------------------------------------------------------- /src/learners/coma_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/learners/coma_learner.py -------------------------------------------------------------------------------- /src/learners/q_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/learners/q_learner.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/main.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/agents/__init__.py -------------------------------------------------------------------------------- /src/modules/agents/rnn_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/agents/rnn_agent.py -------------------------------------------------------------------------------- /src/modules/comm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/__init__.py -------------------------------------------------------------------------------- /src/modules/comm/information_bottleneck_comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/information_bottleneck_comm.py -------------------------------------------------------------------------------- /src/modules/comm/information_bottleneck_comm_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/information_bottleneck_comm_full.py -------------------------------------------------------------------------------- /src/modules/comm/information_bottleneck_comm_not_IB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/information_bottleneck_comm_not_IB.py -------------------------------------------------------------------------------- /src/modules/comm/information_bottleneck_pruned_comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/information_bottleneck_pruned_comm.py -------------------------------------------------------------------------------- /src/modules/comm/tar_comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/comm/tar_comm.py -------------------------------------------------------------------------------- /src/modules/critics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/critics/coma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/critics/coma.py -------------------------------------------------------------------------------- /src/modules/mixers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/mixers/qmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/mixers/qmix.py -------------------------------------------------------------------------------- /src/modules/mixers/vdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/modules/mixers/vdn.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/run.py -------------------------------------------------------------------------------- /src/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/runners/__init__.py -------------------------------------------------------------------------------- /src/runners/episode_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/runners/episode_runner.py -------------------------------------------------------------------------------- /src/runners/episode_runner_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/runners/episode_runner_full.py -------------------------------------------------------------------------------- /src/runners/parallel_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/runners/parallel_runner.py -------------------------------------------------------------------------------- /src/runners/parallel_runner_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/runners/parallel_runner_x.py -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/test.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/test.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/test.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/test.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/test.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/test.sizes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.large.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.large.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.large.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.large.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.large.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.large.sizes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.med.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.med.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.med.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.med.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.med.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.med.sizes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.small.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.small.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.small.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.small.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.small.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.small.sizes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.tiny.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.tiny.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.tiny.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.tiny.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/train.tiny.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/train.tiny.sizes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.colors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/val.colors -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.input.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/val.input.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.labels.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/val.labels.npy -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.query: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/val.shapes -------------------------------------------------------------------------------- /src/shapes/shapes_3x3_single_red/val.sizes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/shapes/shapes_3x3_single_red/val.sizes -------------------------------------------------------------------------------- /src/smac_plus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/__init__.py -------------------------------------------------------------------------------- /src/smac_plus/join1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/join1.py -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/1o_10b_vs_1r.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/1o_10b_vs_1r.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/1o_2r_vs_4r.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/1o_2r_vs_4r.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/3s_vs_5z.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/3s_vs_5z.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/5z_vs_1ul.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/5z_vs_1ul.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/MMM.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/MMM.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/sc2_maps/bane_vs_hM.SC2Map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/sc2_maps/bane_vs_hM.SC2Map -------------------------------------------------------------------------------- /src/smac_plus/starcraft2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/starcraft2/__init__.py -------------------------------------------------------------------------------- /src/smac_plus/starcraft2/maps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/starcraft2/maps/__init__.py -------------------------------------------------------------------------------- /src/smac_plus/starcraft2/starcraft2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/starcraft2/starcraft2.py -------------------------------------------------------------------------------- /src/smac_plus/tracker1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/smac_plus/tracker1.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/dict2namedtuple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/utils/dict2namedtuple.py -------------------------------------------------------------------------------- /src/utils/draw_message_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/utils/draw_message_distributions.py -------------------------------------------------------------------------------- /src/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/utils/logging.py -------------------------------------------------------------------------------- /src/utils/rl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/utils/rl_utils.py -------------------------------------------------------------------------------- /src/utils/timehelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TonghanWang/NDQ/HEAD/src/utils/timehelper.py --------------------------------------------------------------------------------