Update multi_opti notebook

monash-emu · Aug 30, 2023 · f3c7f7b · f3c7f7b
1 parent f3f0a03
commit f3c7f7b
Showing 1 changed file with 5 additions and 26 deletions.
diff --git a/notebooks/user/rragonnet/project_specific/School_Closure/remote/sb_multi_opti.ipynb b/notebooks/user/rragonnet/project_specific/School_Closure/remote/sb_multi_opti.ipynb
@@ -23,34 +23,14 @@
     "len(iso3_list)"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "a31106a5",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "iso3_list = iso3_list[:48]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "d0149785",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "len(iso3_list)"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
    "id": "5337273f",
    "metadata": {},
    "outputs": [],
    "source": [
-    "N_CPUS = 64\n",
+    "N_CPUS = 32\n",
     "N_OPTI_WORKERS = 8\n",
     "assert N_CPUS % N_OPTI_WORKERS == 0"
    ]
@@ -69,7 +49,7 @@
     "\n",
     "    bridge.logger.info(f\"Running optimisation for {len(iso3_list)} countries.\")\n",
     "\n",
-    "    n_parallel_opti_jobs = N_CPUS / N_OPTI_WORKERS\n",
+    "    n_parallel_opti_jobs = 2. * N_CPUS / N_OPTI_WORKERS\n",
     "\n",
     "    multi_country_optimise(\n",
     "        iso3_list=iso3_list, \n",
@@ -104,7 +84,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "analysis_title = \"multi_opti_first48_60dupdate\"\n",
+    "analysis_title = \"multi_opti_15k\"\n",
     "run_path = springboard.launch.get_autumn_project_run_path(\"school_project\", \"multicountry\", analysis_title)\n",
     "run_path"
    ]
@@ -147,7 +127,7 @@
    "outputs": [],
    "source": [
     "# wait function with status printing\n",
-    "print_continuous_status(runner)"
+    "# print_continuous_status(runner)"
    ]
   },
   {
@@ -157,8 +137,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# run_path = 'projects/school_project/multicountry/2023-07-05T2142-multi_opti'\n",
-    "# run_path = 'projects/school_project/multicountry/2023-07-07T1117-multi_opti_first48_60dupdate'"
+    "# run_path = 'projects/school_project/multicountry/2023-08-30T1614-multi_opti_15k'"
    ]
   },
   {