ONNX Model Zoo

Yuquan-Wang/d9_flat-walking-2025-05-27_17-12-59-episode-5000

By Yuquan-Wang · ↓ 0 · ♥ 0 · Updated 2025-06-25

What's in this repo

This page lists the files in Yuquan-Wang/d9_flat-walking-2025-05-27_17-12-59-episode-5000, including policy.onnx.

  • 1 ONNX models (.onnx files)
  • 14 files, including 6 .pt, 2 .pkl, 2 .yaml, 1 .md, 1 .onnx, 1 .0
  • Last updated 2025-06-25

ONNX files

File Size Download
exported/policy.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

IsaacLab Training Output

Model Checkpoints

  • Final model: model_9999.pt

Checkpoint 1000: model_1000.pt- Checkpoint 2000: model_2000.pt- Checkpoint 3000: model_3000.pt- Checkpoint 4000: model_4000.pt- Checkpoint 4999: model_4999.pt

Exported Models

Optimized models for deployment are available in the exported/ directory:

  • policy.pt: Optimized PyTorch model
  • policy.onnx: ONNX format model for cross-platform deployment

Training Configuration

See the params/ directory for training configuration files.

Training Logs

Training logs are available in the logs/ directory.

Demo Videos

Demo videos are available in the videos/ directory.

Usage

Using PyTorch Checkpoints

import torch
model = torch.load('path/to/model.pt')

Using Exported Models

# PyTorch
import torch
model = torch.load('exported/policy.pt')

# ONNX (requires onnxruntime)
import onnxruntime as ort
session = ort.InferenceSession('exported/policy.onnx')

Notes

This version limits the lateral velocity to +- 0.1 m/s.

Model info

tensorboard onnx region:us
↓ 0 ♥ 0

Created: 2025-06-25

Updated: 2025-06-25

View on Hugging Face

Files (14)

.gitattributes
README.md
checkpoints/model_1000.pt
checkpoints/model_2000.pt
checkpoints/model_3000.pt
checkpoints/model_4000.pt
checkpoints/model_4999.pt
exported/policy.onnx ONNX
exported/policy.pt
logs/events.out.tfevents.1748366061.VM-69-2-ubuntu.82096.0
params/agent.pkl
params/agent.yaml
params/env.pkl
params/env.yaml

Related models