ONNX Model Zoo

Yuquan-Wang/d9_rough-walking-2025-06-07_05-23-21-episode-20000

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

What's in this repo

This page lists the files in Yuquan-Wang/d9_rough-walking-2025-06-07_05-23-21-episode-20000, including policy.onnx.

  • 1 ONNX models (.onnx files)
  • 24 files, including 16 .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 2000: model_2000.pt- Checkpoint 5000: model_5000.pt- Checkpoint 8000: model_8000.pt- Checkpoint 9000: model_9000.pt- Checkpoint 10000: model_10000.pt- Checkpoint 11000: model_11000.pt- Checkpoint 12000: model_12000.pt- Checkpoint 13000: model_13000.pt- Checkpoint 14000: model_14000.pt- Checkpoint 15000: model_15000.pt- Checkpoint 16000: model_16000.pt- Checkpoint 17000: model_17000.pt- Checkpoint 18000: model_18000.pt- Checkpoint 19000: model_19000.pt- Checkpoint 19999: model_19999.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 (24)

.gitattributes
README.md
checkpoints/model_10000.pt
checkpoints/model_11000.pt
checkpoints/model_12000.pt
checkpoints/model_13000.pt
checkpoints/model_14000.pt
checkpoints/model_15000.pt
checkpoints/model_16000.pt
checkpoints/model_17000.pt
checkpoints/model_18000.pt
checkpoints/model_19000.pt
checkpoints/model_19999.pt
checkpoints/model_2000.pt
checkpoints/model_5000.pt
checkpoints/model_8000.pt
checkpoints/model_9000.pt
exported/policy.onnx ONNX
exported/policy.pt
logs/events.out.tfevents.1749273831.VM-133-126-ubuntu.321919.0
params/agent.pkl
params/agent.yaml
params/env.pkl
params/env.yaml

Related models