Snowball Fight โ๏ธ, a multi-agent environment for ML-Agents made by Hugging Face
A multi-agent environment using Unity ML-Agents Toolkit where two agents compete in a 1vs1 snowball fight game.
๐ You can play it online at this link.
โ ๏ธ You need to have some skills in ML-Agents if you want to use it if it's not the case check the documentation
The Environment
- Two agents compete in a 1 vs 1 snowball fight game.
- The goal is to hit the opponent team while avoiding the opponent's snowballs โ๏ธ.
Observation Space
Ray-casts:
- 10 ray-casts forward distributed over 100 degrees: detecting opponent.
- 10 ray-casts forward distributed over 100 degrees: detecting walls, shelter and frontier.
- 10 ray-casts forward distributed over 100 degrees: detecting snowballs.
- 3 ray-casts backward distributed over 45 degrees: detecting wall and shelter.
Vector Observations:
- Bool canShoot (you can only shoot a snowball every 2 seconds).
- Float currentHealth: normalized [0, 1]
- Vector3 vertical speed
- Vector3 horizontal speed
- Vector3 "home" position
Action Space (Discrete)
- Vector Action space:
- Four branched actions corresponding to forward, backward, sideways movement, rotation, and snowball shoot.
Agent Reward Function (dependant):
- If the team is injured:
- 0.1 to the shooter.
- If the team is dead:
- (1 - accumulated time penalty): when a snowball hits the opponent, the accumulated time penalty decreases by (1 / MaxStep) every fixed update and is reset to 0 at the beginning of an episode.
- (-1) When a snowball hit our team.
Addendum
- There is no friendly fire, which means that an agent can't shoot himself, or in the future, in a 2vs2 game can't shoot a teammate.
How to use it
Set-up the environment
- Clone this project
git clone https://huggingface.co/ThomasSimonini/ML-Agents-SnowballFight-1vs1
- Open Unity Hub and create a new 3D Project
- In the cloned project folder, open
.\ML-Agents-SnowballFight-1vs1\packages
and copy manifest.json and package.lock.json - Paste these two files in
Your Unity Project\Packages
=> this will install the required packages. - Drop the SnowballFight-1vs1 unity package to your Unity Project.
Watch the trained agents
- If you want to watch the trained agents, open
Assets\1vs1\Scenes\1vs1_v2_Training.
place the\ML-Agents-SnowballFight-1vs1\saved_model\SnowballFight1vs1-4999988.onnx
into BlueAgent and PurpleAgent Model.
Train, the agent
- If you want to train it again, the scene is
Assets\1vs1\Scenes\1vs1_v2_Training.
Training info
- SnowballFight1vs1 was trained with 5100000 steps.
- The final ELO score was 1766.452.
Config File
behaviors: SnowballFight1vs1: trainer_type: ppo hyperparameters: batch_size: 2048 buffer_size: 20480 learning_rate: 0.0003 beta: 0.005 epsilon: 0.2 lambd: 0.95 num_epoch: 3 learning_rate_schedule: constant network_settings: normalize: false hidden_units: 512 num_layers: 2 vis_encode_type: simple reward_signals: extrinsic: gamma: 0.99 strength: 1.0 keep_checkpoints: 40 checkpoint_interval: 200000 max_steps: 50000000 time_horizon: 1000 summary_freq: 50000 self_play: save_steps: 50000 team_change: 200000 swap_steps: 2000 window: 10 play_against_latest_model_ratio: 0.5 initial_elo: 1200.0
- Downloads last month
- 9