use correct tf model weights filename (#2901)

NVIDIA · Sep 4, 2024 · d07b3f6 · d07b3f6
1 parent 81f23fc
commit d07b3f6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/hello-world/hello_world.ipynb b/examples/hello-world/hello_world.ipynb
@@ -567,7 +567,7 @@
     "common_decomposers.register()\n",
     "\n",
     "result = sess.download_job_result(job_id)\n",
-    "with open(result + \"/workspace/app_server/tf2weights.fobs\", \"rb\") as f:\n",
+    "with open(result + \"/workspace/app_server/tf_model.weights.h5\", \"rb\") as f:\n",
     "    bytes = f.read()\n",
     "\n",
     "weights = fobs.loads(bytes)\n",
@@ -878,7 +878,7 @@
     "\n",
     "common_decomposers.register()\n",
     "result = sess.download_job_result(job_id)\n",
-    "with open(result + \"/workspace/app_server/tf2weights.fobs\", \"rb\") as f:\n",
+    "with open(result + \"/workspace/app_server/tf_model.weights.h5\", \"rb\") as f:\n",
     "    bytes = f.read()\n",
     "\n",
     "weights = fobs.loads(bytes)\n",
@@ -971,4 +971,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}