Skip to content

Commit

Permalink
feat: Add training script for HAPPO algorithm
Browse files Browse the repository at this point in the history
  • Loading branch information
ch33nchan committed Dec 27, 2024
1 parent a954780 commit 2dd8094
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions scripts/train_happo.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
import jax
from mava.algorithms.happo import HAPPO
from mava.configs.happo_config import HAPPOConfig
from mava.environments import create_environment
from mava.trainers import Trainer

def main():
config = HAPPOConfig()
environment = create_environment(config)
algorithm = HAPPO(config)
trainer = Trainer(config, environment, algorithm)
trainer.train()

if __name__ == "__main__":
main()

0 comments on commit 2dd8094

Please sign in to comment.