Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

DOCS-2668: Update upload-training-script.md #3143

Merged
merged 2 commits into from
Jul 19, 2024
Merged

DOCS-2668: Update upload-training-script.md #3143

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ab2fd79
DOCS-2668: Update upload-training-script.md
npentrel Jul 18, 2024
556b067
Update upload-training-script.md
npentrel Jul 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions docs/services/ml/upload-training-script.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,7 @@ When you submit a training job with this training script, this function saves th
Once the training job is complete, Viam looks at that directory and creates a package with all of the contents of the directory, creating or updating a registry item for the ML model.

{{% /expand%}}
{{%expand "Step 4: Write main function" %}}
{{%expand "Step 5: Write main function" %}}

Now, write all the code that runs the training job invoking the previously defined helper functions.
Write this into the top level code of <file>training.py</file>, which is executed when the file runs as a script.
Expand Down Expand Up @@ -389,7 +389,7 @@ You can use the [`viam training-script upload`](/cli/#training-script) command t

For example:

```sh {class="line-numbers linkable-line-numbers"}
```sh {class="command-line" data-prompt="$"}
viam training-script upload --path=<path-to-tar.gz> --org-id=<INSERT ORG ID> --script-name="MyCustomTrainingScript"
```

Expand All @@ -411,7 +411,7 @@ For example:
{{< tabs >}}
{{% tab name="from-registry" %}}

```sh {class="line-numbers linkable-line-numbers"}
```sh {class="command-line" data-prompt="$"}
viam train submit custom from-registry --dataset-id=<INSERT DATASET ID> --org-id=<INSERT ORG ID> --model-name="MyRegistryModel" --model-version="2" --version="1" --path=<path-to-tar.gz> --script-name="MyCustomTrainingScript"
```

Expand All @@ -420,7 +420,7 @@ This command submits a training job to the previously uploaded `"MyCustomTrainin
{{% /tab %}}
{{% tab name="with-upload" %}}

```sh {class="line-numbers linkable-line-numbers"}
```sh {class="command-line" data-prompt="$"}
viam train submit custom with-upload --dataset-id=<INSERT DATASET ID> --model-org-id=<INSERT ORG ID> --model-name="MyRegistryModel" --model-type="single_label_classification" --model-version="2" --version="1" --path=<path-to-tar.gz> --script-name="MyCustomTrainingScript"
```

Expand Down
Loading