Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file added example/python/exported/policy.onnx
Binary file not shown.
Binary file added example/python/exported/policy.pt
Binary file not shown.
7 changes: 7 additions & 0 deletions example/python/exported/read.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
import onnx

# ONNX 모델 열기
model = onnx.load("policy.onnx")

# 모델 구조 출력
print(onnx.helper.printable_graph(model.graph))
Binary file added example/python/params/agent.pkl
Binary file not shown.
49 changes: 49 additions & 0 deletions example/python/params/agent.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
seed: 42
device: cuda:0
num_steps_per_env: 24
max_iterations: 300
empirical_normalization: null
obs_groups: {}
clip_actions: null
save_interval: 50
experiment_name: unitree_go2_flat
run_name: ''
logger: tensorboard
neptune_project: isaaclab
wandb_project: isaaclab
resume: false
load_run: .*
load_checkpoint: model_.*.pt
class_name: OnPolicyRunner
policy:
class_name: ActorCritic
init_noise_std: 1.0
noise_std_type: scalar
actor_obs_normalization: false
critic_obs_normalization: false
actor_hidden_dims:
- 128
- 128
- 128
critic_hidden_dims:
- 128
- 128
- 128
activation: elu
algorithm:
class_name: PPO
num_learning_epochs: 5
num_mini_batches: 4
learning_rate: 0.001
schedule: adaptive
gamma: 0.99
lam: 0.95
entropy_coef: 0.01
desired_kl: 0.01
max_grad_norm: 1.0
value_loss_coef: 1.0
use_clipped_value_loss: true
clip_param: 0.2
normalize_advantage_per_mini_batch: false
rnd_cfg: null
symmetry_cfg: null
Binary file added example/python/params/env.pkl
Binary file not shown.
Loading