Update train_model.ipynb

truefoundry · Feb 12, 2024 · bed6314 · bed6314
1 parent 820c922
commit bed6314
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mnist-classifaction/train_model.ipynb b/mnist-classifaction/train_model.ipynb
@@ -91,7 +91,7 @@
       "source": [
        "import mlfoundry\n",
        "\n",
-       "client = mlfoundry.get_client()"
+       "client = mlfoundry.get_client(disable_analytics=True)"
       ]
      },
      {
@@ -163,7 +163,7 @@
       "outputs": [],
       "source": [
        "client.create_ml_repo(ML_REPO_NAME)\n",
-       "run = client.create_run(ml_repo=ML_REPO_NAME, name=\"train-model\")\n",
+       "run = client.create_run(ml_repo=ML_REPO_NAME, run_name=\"train-model\")\n",
        "\n",
        "# Plot some sample images\n",
        "plt.figure(figsize=(10, 5))\n",
@@ -350,4 +350,4 @@
     "nbformat": 4,
     "nbformat_minor": 4
    }
-
+