We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0cc7be4 commit 9993824Copy full SHA for 9993824
1 file changed
notebooks/community/model_garden/model_garden_pytorch_llama3_1_finetuning.ipynb
@@ -705,8 +705,11 @@
705
" eval_machine_type = \"g2-standard-24\"\n",
706
" eval_accelerator_type = \"NVIDIA_L4\"\n",
707
" eval_accelerator_count = 2\n",
708
- " dws_kwargs = {}\n",
709
- " is_dynamic_workload_scheduler = False\n",
+ " dws_kwargs = {\n",
+ " \"max_wait_duration\": 10800, # 180 minutes\n",
710
+ " \"scheduling_strategy\": gca_custom_job_compat.Scheduling.Strategy.FLEX_START,\n",
711
+ " }\n",
712
+ " is_dynamic_workload_scheduler = True\n",
713
"elif \"70b\" in base_model_id.lower():\n",
714
" eval_machine_type = \"a2-ultragpu-4g\"\n",
715
" eval_accelerator_type = \"NVIDIA_A100_80GB\"\n",
0 commit comments