How do I run my custom model training from vscode

[nitya_r]

I have run a custom model training on workbench. I want to execute this training from VSCode. Is that possible and if yes, how do i do it?

[Aris_O]

Hi @nitya_r 

Welcome back to Google Cloud Community.

It is possible to execute training to VSCode. Connect to a Custom User Managed Notebook

To use VSCode with a Vertex AI Workbench obtained starting from a custom container (e.g. our custom-image machine) a few additional steps are required:

• Connect to the instance following the same procedure described in the previous section.If you check /home/jupyter directory you will find the cloned content, but you will not find libraries, environments or code configured on your container.
• In fact, your custom container is executed on a docker instance on the virtual machine.
• To be able to connect to the container with VSCode, your user must belong to the docker group of the remote machine. Add your user as follows.
  
  This reference might help you:
  https://medium.com/google-cloud/choose-the-ide-you-want-and-develop-on-vertex-ai-workbench-part-i-94...

[irismo]

Hey Aris, thanks for your reply. I got the similar issue here, but I'm only allowed to create managed notebooks instead of compute instances or user-managed notebooks. Is there a way to connect VS code with managed notebooks?

[pritamdodeja]

@nitya_r did you use any pipelines like tfx etc., or was your ingestion, preprocessing, training loop all in tf/keras?

[hsbedi]

Hello!

Please follow the instructions in the below link to setup Application Default Credentials (ADC) on your dev box. Code run within VScode will then use these credentials.

https://cloud.google.com/docs/authentication/provide-credentials-adc#local-dev

Thanks!