搭配參數執行筆記本
Vertex AI Workbench 代管筆記本執行個體可讓您在筆記本執行作業中使用參數值,指定筆記本檔案執行程式碼的不同方式。本頁面說明如何設定筆記本檔案以使用參數,以及如何執行指定筆記本參數的不同值。
使用參數執行筆記檔案的不同版本
您可以在執行作業中使用 Notebook 參數值,以便在指定差異時執行相同的 Notebook 程式碼,例如:
指定要使用的其他資料集,或資料集的其他樣本大小。
指定不同的模型設定,例如學習率或最佳化器類型。
執行不同的模型,或執行同一模型的不同版本。
如何在執行筆記本時使用參數
執行含有參數的 Notebook 的程序有兩個主要步驟:
在其中一個筆記本檔案的儲存格中加入
parameters
標記。雖然這不是技術要求,但這個儲存格通常會包含為參數變數指派值的程式碼,但這不是技術要求。如果您未在執行作業中指派不同的參數值,執行作業會使用 Notebook 檔案中的參數值做為預設值。為筆記型電腦檔案建立執行作業,其中包含參數的新值。請使用以下模式設定參數和參數值的格式:
parameter1=value1,parameter2=value2
。這個格式要求在參數值組之間使用逗號,且不含空格和引號。執行時,執行程式會在記事本中新增一個儲存格,直接在標示為parameters
的儲存格後方更新參數值。
事前準備
- Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
-
Make sure that billing is enabled for your Google Cloud project.
-
Enable the Notebooks and Vertex AI APIs.
-
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
-
Make sure that billing is enabled for your Google Cloud project.
-
Enable the Notebooks and Vertex AI APIs.
- 如果您尚未建立代管型筆記本執行個體,請先完成這項作業。
-
筆記本檢視器 (
roles/notebooks.viewer
) -
Vertex AI 使用者 (
roles/aiplatform.user
) -
Storage 管理員 (
roles/storage.admin
) 在代管型筆記本執行個體的 JupyterLab 使用者介面中,開啟要執行的筆記本檔案。
在一個儲存格中編寫程式碼,為參數變數指派值。如果您在執行時未指派不同的參數值,則這些值會是 Notebook 檔案使用的值。
確認您仍選取了參數儲存格,然後在右側邊欄中按一下
「Property inspector」。在屬性檢視器的「Cell Tags」部分中,按一下「Add Tag」,輸入
parameters
,然後按下Enter
。在代管型筆記本執行個體的 JupyterLab 使用者介面中,按一下
「Executor」 按鈕。在「提交 Notebook 至執行程式」對話方塊中,在「執行名稱」欄位中輸入執行作業的名稱。
選取「Machine type」(機器類型)和「Accelerator type」(加速器類型)。
選取環境。
在「Type」欄位中,選取「One-time execution」或「Schedule-based recurring executions」,然後完成排程執行作業的對話方塊。
在「進階選項」中,選取要執行筆記本的「區域」。
在「Cloud Storage bucket」欄位中,選取可用的 Cloud Storage bucket,或輸入新 bucket 的名稱,然後按一下「Create and select」。執行程式會將筆記本輸出內容儲存在這個 Cloud Storage 值區中。
在「Notebook 參數化」部分和「輸入參數」文字方塊中,新增以半形逗號分隔的 Notebook 參數,例如
optimizer=SGD,learning_rate=0.01
。這個格式不含空格和引號。設定執行作業的其餘部分,然後按一下「提交」。
- 進一步瞭解如何在執行緒中執行 Notebook 程式碼。
必要的角色
為確保您的執行個體服務帳戶具備與 Vertex AI Workbench 執行程序互動的必要權限,請要求管理員將專案的下列 IAM 角色授予您的執行個體服務帳戶:
如要進一步瞭解如何授予角色,請參閱「管理專案、資料夾和機構的存取權」。
管理員也可能會透過自訂角色或其他預先定義的角色,為執行個體的服務帳戶授予必要權限。
開啟 JupyterLab
如要開啟 JupyterLab 並準備要執行的 Notebook 檔案,請完成下列步驟。