Executar execuções de notebook com parâmetros

As instâncias de notebooks gerenciados do Vertex AI Workbench permitem usar valores de parâmetro nas execuções de notebook para especificar diferenças na forma como o código do arquivo de notebook é executado. Nesta página, você verá como configurar o arquivo de notebook para usar parâmetros e como gerar execuções que especificam valores diferentes para os parâmetros de notebook.

Use parâmetros para executar iterações diferentes do arquivo de notebook

É possível usar os valores de parâmetro do notebook nas execuções para executar o mesmo código do notebook, especificando diferenças como as seguintes:

  • Especifique um conjunto de dados diferente ou um tamanho de amostra diferente para ser usado.

  • Especifique diferentes configurações de modelo, como taxa de aprendizado ou tipo de otimizador.

  • Execute modelos diferentes ou versões diferentes do mesmo modelo.

Como usar parâmetros em uma execução de notebook

O processo para executar um notebook com parâmetros tem duas etapas principais:

  1. Adicione a tag parameters a uma das células do arquivo de notebook. Embora não seja um requisito técnico, essa célula normalmente contém código que atribui valores a suas variáveis de parâmetro. Se você não atribuir valores de parâmetro diferentes na sua execução, ela usará os valores de parâmetro no arquivo do notebook como valores padrão.

  2. Crie uma execução para o arquivo de notebook que inclua novos valores para os parâmetros. Use o padrão a seguir para formatar os parâmetros e os valores deles: parameter1=value1,parameter2=value2. O formato requer vírgulas entre pares de parâmetros/valores, sem espaços e aspas. Quando a execução é gerada, o executor adiciona uma célula ao notebook que atualiza os valores dos parâmetros logo após a célula marcada como parameters.

Antes de começar

  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.

  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.
  8. Crie uma instância de notebook gerenciado se ainda não tiver feito isso.

Funções exigidas

Para garantir que a conta de serviço da instância tenha as permissões necessárias para interagir com o executor do Vertex AI Workbench, peça ao administrador para conceder a ela os seguintes papéis do IAM no projeto:

Para mais informações sobre a concessão de papéis, consulte Gerenciar o acesso a projetos, pastas e organizações.

O administrador também pode conceder à conta de serviço da instância as permissões necessárias por meio de papéis personalizados ou de outros papéis predefinidos.

Abrir JupyterLab

Para abrir o JupyterLab e preparar um arquivo de notebook para ser executado, conclua as etapas a seguir.

  1. Abrir o JupyterLab.

  2. Faça upload de um arquivo de notebook (ipynb), abra um atual ou abra um novo arquivo de notebook e adicione o código que você quer executar no novo notebook.

  3. Verifique se o código do arquivo do notebook atende aos requisitos de uso do executor.

Adicionar a tag parameters a uma célula do notebook

  1. Na interface do usuário do JupyterLab da instância de notebooks gerenciados, abra o arquivo de notebook que você quer executar.

  2. Escreva o código em uma célula que atribua valores às suas variáveis de parâmetro. Esses são os valores que o arquivo de notebook usará se você não atribuir valores de parâmetro diferentes na execução.

  3. Verifique se a célula de parâmetros ainda está selecionada e, na barra lateral direita, clique no  Inspetor de propriedades.

  4. No inspetor de propriedade, na seção Tags de célula, clique em Adicionar tag, digite parameters e pressione Enter.

Fornecer valores de parâmetro para a execução

  1. Na interface do usuário do JupyterLab da instância de notebooks gerenciados, clique no botão Executor .

  2. Na caixa de diálogo Enviar notebooks para o Executor, insira um nome para sua execução no campo Nome da execução.

  3. Selecione um tipo de máquina e um tipo de acelerador.

  4. Selecione um Ambiente.

  5. No campo Tipo, selecione Execução única ou Execuções recorrentes baseadas em programação e complete a caixa de diálogo para programar execuções.

  6. Em Opções avançadas, selecione a Região em que você quer executar o notebook.

  7. No campo bucket do Cloud Storage, selecione um bucket disponível ou digite um nome para um novo bucket e clique em Criar e selecionar. O executor armazena a saída do notebook nesse bucket do Cloud Storage.

  8. Na seção Parametrização do notebook, na caixa de texto Parâmetros de entrada, adicione os parâmetros do notebook separados por vírgulas, por exemplo, optimizer=SGD,learning_rate=0.01. O formato requer a falta de espaços e aspas.

  9. Configure o restante da execução e clique em Enviar.

Notebook

A seguir