搭配參數執行筆記本

Vertex AI Workbench 代管筆記本執行個體可讓您在筆記本執行作業中使用參數值,指定筆記本檔案執行程式碼的不同方式。本頁面說明如何設定筆記本檔案以使用參數,以及如何執行指定筆記本參數的不同值。

使用參數執行筆記檔案的不同版本

您可以在執行作業中使用 Notebook 參數值,以便在指定差異時執行相同的 Notebook 程式碼,例如:

  • 指定要使用的其他資料集,或資料集的其他樣本大小。

  • 指定不同的模型設定,例如學習率或最佳化器類型。

  • 執行不同的模型,或執行同一模型的不同版本。

如何在執行筆記本時使用參數

執行含有參數的 Notebook 的程序有兩個主要步驟:

  1. 在其中一個筆記本檔案的儲存格中加入 parameters 標記。雖然這不是技術要求,但這個儲存格通常會包含為參數變數指派值的程式碼,但這不是技術要求。如果您未在執行作業中指派不同的參數值,執行作業會使用 Notebook 檔案中的參數值做為預設值。

  2. 為筆記型電腦檔案建立執行作業,其中包含參數的新值。請使用以下模式設定參數和參數值的格式:parameter1=value1,parameter2=value2。這個格式要求在參數值組之間使用逗號,且不含空格和引號。執行時,執行程式會在記事本中新增一個儲存格,直接在標示為 parameters 的儲存格後方更新參數值。

事前準備

  1. 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.
  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  3. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Notebooks and Vertex AI APIs.

    Enable the APIs

  5. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  6. Make sure that billing is enabled for your Google Cloud project.

  7. Enable the Notebooks and Vertex AI APIs.

    Enable the APIs

  8. 如果您尚未建立代管型筆記本執行個體,請先完成這項作業。
  9. 必要的角色

    為確保您的執行個體服務帳戶具備與 Vertex AI Workbench 執行程序互動的必要權限,請要求管理員將專案的下列 IAM 角色授予您的執行個體服務帳戶:

    如要進一步瞭解如何授予角色,請參閱「管理專案、資料夾和機構的存取權」。

    管理員也可能會透過自訂角色或其他預先定義的角色,為執行個體的服務帳戶授予必要權限。

    開啟 JupyterLab

    如要開啟 JupyterLab 並準備要執行的 Notebook 檔案,請完成下列步驟。

    1. 開啟 JupyterLab

    2. 上傳筆記本 (ipynb) 檔案、開啟現有檔案,或開啟新筆記本檔案,然後將要執行的程式碼新增至新筆記本。

    3. 確認您的筆記本檔案程式碼符合使用執行緒的規定

    在筆記本儲存格中加入 parameters 標記

    1. 在代管型筆記本執行個體的 JupyterLab 使用者介面中,開啟要執行的筆記本檔案。

    2. 在一個儲存格中編寫程式碼,為參數變數指派值。如果您在執行時未指派不同的參數值,則這些值會是 Notebook 檔案使用的值。

    3. 確認您仍選取了參數儲存格,然後在右側邊欄中按一下 「Property inspector」

    4. 在屬性檢視器的「Cell Tags」部分中,按一下「Add Tag」,輸入 parameters,然後按下 Enter

    為執行作業提供參數值

    1. 在代管型筆記本執行個體的 JupyterLab 使用者介面中,按一下 「Executor」 按鈕。

    2. 在「提交 Notebook 至執行程式」對話方塊中,在「執行名稱」欄位中輸入執行作業的名稱。

    3. 選取「Machine type」(機器類型)和「Accelerator type」(加速器類型)

    4. 選取環境

    5. 在「Type」欄位中,選取「One-time execution」或「Schedule-based recurring executions」,然後完成排程執行作業的對話方塊。

    6. 在「進階選項」中,選取要執行筆記本的「區域」

    7. 在「Cloud Storage bucket」欄位中,選取可用的 Cloud Storage bucket,或輸入新 bucket 的名稱,然後按一下「Create and select」。執行程式會將筆記本輸出內容儲存在這個 Cloud Storage 值區中。

    8. 在「Notebook 參數化」部分和「輸入參數」文字方塊中,新增以半形逗號分隔的 Notebook 參數,例如 optimizer=SGD,learning_rate=0.01。這個格式不含空格和引號。

    9. 設定執行作業的其餘部分,然後按一下「提交」

    筆記本

    後續步驟