jarski/Worm-Unity-ML
By jarski · ↓ 0 · ♥ 0 · Updated 2024-08-02
What's in this repo
This page lists the files in jarski/Worm-Unity-ML, including Worm.onnx, Worm-0.onnx, Worm-1010000.onnx and more.
- 5 ONNX models (.onnx files)
- 19 files, including 5 .onnx, 5 .pt, 3 .json, 2 .0, 1 .md, 1 .yaml
- Last updated 2024-08-02
ONNX files
| File | Size | Download |
|---|---|---|
| Worm.onnx | — | Download |
| Worm/Worm-0.onnx | — | Download |
| Worm/Worm-1010000.onnx | — | Download |
| Worm/Worm-499000.onnx | — | Download |
| Worm/Worm-999000.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
ppo Agent playing Worm
This is a trained model of a ppo agent playing Worm using the Unity ML-Agents Library.
Usage (with ML-Agents)
The Documentation: https://unity-technologies.github.io/ml-agents/ML-Agents-Toolkit-Documentation/
We wrote a complete tutorial to learn to train your first agent using ML-Agents and publish it to the Hub:
- A short tutorial where you teach Huggy the Dog 🐶 to fetch the stick and then play with him directly in your browser: https://huggingface.co/learn/deep-rl-course/unitbonus1/introduction
- A longer tutorial to understand how works ML-Agents: https://huggingface.co/learn/deep-rl-course/unit5/introduction
Resume the training
mlagents-learn <your_configuration_file_path.yaml> --run-id=<run_id> --resume
Watch your Agent play
You can watch your agent playing directly in your browser
- If the environment is part of ML-Agents official environments, go to https://huggingface.co/unity
- Step 1: Find your model_id: jarski/Worm-Unity-ML
- Step 2: Select your .nn /.onnx file
- Click on Watch the agent play 👀
Model info
Created: 2024-08-02
Updated: 2024-08-02
View on Hugging Face