chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix embarrassing bug with ckpt.

Browse Source 
PiperOrigin-RevId: 686679906
Change-Id: I61126ce94214b5f7b25b31e343a243d1a1dbb9f4
This commit is contained in:
Baruch Tabanpour
2024-10-16 16:23:03 -07:00
committed by Copybara-Service
parent ffdb800e45
commit 43a7493d17
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/tutorial.ipynb
+2 -2
View File
@@ -1513,8 +1513,8 @@
"source": [ "source": [
"# grab the latest checkpoint from the flat terrain joystick policy\n", "# grab the latest checkpoint from the flat terrain joystick policy\n",
"latest_ckpts = list(ckpt_path.glob('*'))\n", "latest_ckpts = list(ckpt_path.glob('*'))\n",
"latest_ckpts.sort()\n", "latest_ckpts.sort(key=lambda x: int(x.as_posix().split('/')[-1]))\n",
"latest_ckpt = latest_ckpts[0]\n", "latest_ckpt = latest_ckpts[-1]\n",
"\n", "\n",
"train_fn = functools.partial(\n", "train_fn = functools.partial(\n",
" ppo.train, num_timesteps=40_000_000, num_evals=5,\n", " ppo.train, num_timesteps=40_000_000, num_evals=5,\n",