ONNX Model Zoo

hivex-research/hivex-AWS-PPO-baseline-task-0-difficulty-6

By hivex-research · ↓ 0 · ♥ 0 · Updated 2025-03-20

What's in this repo

This page lists the files in hivex-research/hivex-AWS-PPO-baseline-task-0-difficulty-6, including Agent.onnx, Agent-1498956.onnx, Agent-1800054.onnx and more.

  • 5 ONNX models (.onnx files)
  • 16 files, including 5 .onnx, 5 .pt, 2 .json, 1 .0, 1 .md, 1 .yaml
  • Last updated 2025-03-20

ONNX files

File Size Download
Agent.onnx — Download
Agent/Agent-1498956.onnx — Download
Agent/Agent-1800054.onnx — Download
Agent/Agent-497978.onnx — Download
Agent/Agent-999906.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

This model serves as the baseline for the Aerial Wildfire Suppression environment, trained and tested on task 0 with difficulty 6 using the Proximal Policy Optimization (PPO) algorithm.

Environment: Aerial Wildfire Suppression

Task: 0

Difficulty: 6

Algorithm: PPO

Episode Length: 3000

Training max_steps: 1800000

Testing max_steps: 180000

Train & Test Scripts

Download the Environment

Model info

hivex tensorboard onnx hivex-aerial-wildfire-suppression reinforcement-learning multi-agent-reinforcement-learning arxiv:2501.04180 model-index region:us
↓ 0 ♥ 0

Created: 2024-08-30

Updated: 2025-03-20

View on Hugging Face

Files (16)

.gitattributes
Agent.onnx ONNX
Agent/Agent-1498956.onnx ONNX
Agent/Agent-1498956.pt
Agent/Agent-1800054.onnx ONNX
Agent/Agent-1800054.pt
Agent/Agent-497978.onnx ONNX
Agent/Agent-497978.pt
Agent/Agent-999906.onnx ONNX
Agent/Agent-999906.pt
Agent/checkpoint.pt
Agent/events.out.tfevents.1717381318.RICHARD.24884.0
README.md
configuration.yaml
run_logs/timers.json
run_logs/training_status.json

Related models