Fix embarrassing bug with ckpt.

PiperOrigin-RevId: 686679906 Change-Id: I61126ce94214b5f7b25b31e343a243d1a1dbb9f4
2024-10-16 16:23:03 -07:00
parent ffdb800e45
commit 43a7493d17
1 changed files with 2 additions and 2 deletions
@@ -1513,8 +1513,8 @@
   "source": [
    "# grab the latest checkpoint from the flat terrain joystick policy\n",
    "latest_ckpts = list(ckpt_path.glob('*'))\n",
-    "latest_ckpts.sort()\n",
-    "latest_ckpt = latest_ckpts[0]\n",
+    "latest_ckpts.sort(key=lambda x: int(x.as_posix().split('/')[-1]))\n",
+    "latest_ckpt = latest_ckpts[-1]\n",
    "\n",
    "train_fn = functools.partial(\n",
    "      ppo.train, num_timesteps=40_000_000, num_evals=5,\n",