Update raytune_pytorch_cnn.ipynb

facebook · Nov 10, 2023 · c4028ea · c4028ea
1 parent 82f6f88
commit c4028ea
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tutorials/raytune_pytorch_cnn.ipynb b/tutorials/raytune_pytorch_cnn.ipynb
@@ -191,7 +191,8 @@
       },
       "source": [
         "## 4. Run optimization\n",
-        "Execute the Ax optimization and trial evaluation in RayTune using [AxSearch algorithm](https://ray.readthedocs.io/en/latest/tune-searchalg.html#ax-search):"
+        "Execute the Ax optimization and trial evaluation in RayTune using [AxSearch algorithm](https://ray.readthedocs.io/en/latest/tune-searchalg.html#ax-search):\n"
+        "We only run 10 trials for demonstration. It is generally recommended to run more trials for best results."
       ]
     },
     {
@@ -211,7 +212,7 @@
         "algo = tune.search.ConcurrencyLimiter(algo, max_concurrent=3)\n",
         "tune.run(\n",
         "    train_evaluate,\n",
-        "    num_samples=30,\n",
+        "    num_samples=10,\n",
         "    search_alg=algo,\n",
         "    verbose=0,  # Set this level to 1 to see status updates and to 2 to also see trial results.\n",
         "    # To use GPU, specify: resources_per_trial={\"gpu\": 1}.\n",