# Multi-agent Coordination for Tracking

## Environment: Pose
* v1: Fully observation with hierarchy
* v2: Partial observation with hierarchy
* v3: v2 with obstacles with hierarchy
## Requirements

To install requirements:

```setup
pip install -r requirements.txt
```

## Training
To train, run this command:

```train
python main.py --env Pose-v3 --model ToM-v5 --workers 6 --norm-reward
```

## Evaluation

To evaluate my model, run:

```eval
python main.py --env Pose-v3 --model ToM-v5 --workers 0 --load-model-dir trainedModel/model.pth
```

You can use trained models directly from the folder "trainedModel". Please wait for a few moments before the test process finishes 20 episodes and calculates the average episode reward.
