Skip to content

A reinforcement learning cross attention channel with centralized training and execution for NMMO NeurIPS 2023

License

Notifications You must be signed in to change notification settings

saidineshpola/nmmo-rl

Repository files navigation

Running the Script

To run the train.py script, navigate to the directory containing the script and run the following command with Config from reinforcement_learning/config.py:

python train.py

figure

icon Welcome to the Platform!

Experiment Tracking

  • Task Completion Plot

To track the progress of our experiments and view detailed metrics, visit our experiment tracking dashboard on Weights & Biases.

Results

In the "results" folder, you can find the following images:

  1. Visual Tile for Each Agent in NMMO Grid:

    • Tile Encoder
    • Description: This image represents the visual tile for each agent in the Neural MMO grid, providing insights into the spatial representation of the environment.
  2. Action Decoder of NMMO:

    • Action Decoder
    • Description: This is the action decoder architecture of the Neural MMO (NMMO) model.
  3. Multihead Attention Communication Channel between Agents:

    • Multihead Attention
    • Description: This is the communication channel between agents using multihead attention mechanism.

Credits

  1. The foundational work and inspiration for this project came from the nmmo baseline
  2. We drew multi head attention communication insights and ideas from the research paper Is Centralized Training with Decentralized Execution Framework Centralized Enough for MARL?

About

A reinforcement learning cross attention channel with centralized training and execution for NMMO NeurIPS 2023

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published