hivex-research/hivex-WRM-PPO-baseline-task-2-difficulty-1

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

What's in this repo

This page lists the files in hivex-research/hivex-WRM-PPO-baseline-task-2-difficulty-1, including Agent.onnx, Agent-451800.onnx.

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

ONNX files

File	Size	Download
Agent.onnx	—	Download
Agent/Agent-451800.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 Wildfire Resource Management environment, trained and tested on task 2 with difficulty 1 using the Proximal Policy Optimization (PPO) algorithm.

Environment: Wildfire Resource Management

Task: 2

Difficulty: 1

Algorithm: PPO

Episode Length: 500

Training max_steps: 450000

Testing max_steps: 45000

Train & Test Scripts

Download the Environment

Model info

reinforcement-learning hivex

hivex tensorboard onnx hivex-wildfire-resource-management reinforcement-learning multi-agent-reinforcement-learning arxiv:2501.04180 model-index region:us

↓ 0 ♥ 0

Created: 2024-08-29

Updated: 2025-03-20

View on Hugging Face

Files (10)

.gitattributes

Agent.onnx ONNX

Agent/Agent-451800.onnx ONNX

Agent/Agent-451800.pt

Agent/checkpoint.pt

Agent/events.out.tfevents.1716691862.RICHARD.11632.0

README.md

configuration.yaml

run_logs/timers.json

run_logs/training_status.json

Related models

More reinforcement-learning models

Browse all reinforcement-learning models