Views
No views yet
exported/ directory:policy.pt: Optimized PyTorch modelpolicy.onnx: ONNX format model for cross-platform deploymentparams/ directory for training configuration files.logs/ directory.videos/ directory.1import torch
2model = torch.load('path/to/model.pt')1# PyTorch
2import torch
3model = torch.load('exported/policy.pt')
4
5# ONNX (requires onnxruntime)
6import onnxruntime as ort
7session = ort.InferenceSession('exported/policy.onnx')