From f719274ef6398ce917152cc8860c44544d6995f4 Mon Sep 17 00:00:00 2001 From: Wesley Maa <57124298+WT-MM@users.noreply.github.com> Date: Sun, 29 Sep 2024 13:25:05 -0700 Subject: [PATCH] ONNX ppo inference on MuJoCo wasm (#423) * copy nathan code * add dora * stompypro model * local wasm ppo * checkpoint * onnx ppo * remove sim * add packages * pawel fixes * remove packages * center div * lint * remove extraneous files * remove extra slash * fixed build? --------- Co-authored-by: Winston Hsiao <96440583+Winston-Hsiao@users.noreply.github.com> --- .gitignore | 5 +- frontend/index.html | 3 + frontend/package-lock.json | 1039 +++- frontend/package.json | 5 +- frontend/public/examples/main.js | 487 +- frontend/public/examples/models/standing.onnx | Bin 0 -> 1924559 bytes frontend/public/examples/mujocoUtils.js | 2 +- frontend/public/examples/ppo.js | 45 + frontend/public/examples/scenes/stompypro.xml | 41 +- .../examples/jsm/controls/DragControls.js | 220 - .../examples/jsm/controls/OrbitControls.js | 1295 ----- .../examples/jsm/libs/lil-gui.module.min.js | 8 - .../src/components/listing/mujoco/index.tsx | 1 - frontend/src/components/pages/MuJoCoTest.tsx | 2 +- frontend/src/components/pages/TestPage.tsx | 13 + frontend/src/gen/api.ts | 4952 +++++++++-------- 16 files changed, 3870 insertions(+), 4248 deletions(-) create mode 100644 frontend/public/examples/models/standing.onnx create mode 100644 frontend/public/examples/ppo.js delete mode 100644 frontend/public/node_modules/three/examples/jsm/controls/DragControls.js delete mode 100644 frontend/public/node_modules/three/examples/jsm/controls/OrbitControls.js delete mode 100644 frontend/public/node_modules/three/examples/jsm/libs/lil-gui.module.min.js create mode 100644 frontend/src/components/pages/TestPage.tsx diff --git a/.gitignore b/.gitignore index 04aa77f9..c8d12c71 100644 --- a/.gitignore +++ b/.gitignore @@ -30,6 +30,5 @@ out*/ !.elasticbeanstalk/*.cfg.yml !.elasticbeanstalk/*.global.yml -# mujoco wasm setup -!/public/dist -!/public/node_modules/three/build +# .DS_Store +.DS_Store diff --git a/frontend/index.html b/frontend/index.html index 81185a33..8a687ef6 100644 --- a/frontend/index.html +++ b/frontend/index.html @@ -37,6 +37,9 @@ + + +