You're looking at a specific version of this model. Jump to the model overview.
phospho-app /gr00t-policy:92ae1c0f
Input schema
The fields you can use to run this model with an API. If you don’t give a value for a field its default value will be used.
| Field | Type | Default value | Description |
|---|---|---|---|
| dataset_repo_id |
string
|
Hugging Face dataset ID to train on, LeRobot format > v2.0 expected, i.e. 'LegrandFrederic/dual-setup'
|
|
| hugging_face_token |
string
|
Hugging Face API token, needed to download your dataset and upload your model. Please use a fine grained token with Repository write access.
|
|
| wandb_api_key |
string
|
Weights & Biases API key (optional, to track the online training), find yours here: https://wandb.ai/authorize
|
|
| hf_model_name |
string
|
|
Hugging Face model name to upload the trained model to (optional). If not provided, a random name will be generated.
|
| batch_size |
integer
|
64
Min: 1 Max: 80 |
Batch size for training
|
| number_of_epochs_to_train_for |
integer
|
20
Min: 1 Max: 50 |
Number of epochs to train for
|
| learning_rate |
number
|
0.0002
Min: 0.00001 Max: 0.01 |
Learning rate for training
|
| modal_token_id |
string
|
Modal token ID, if you don't know what this is, leave it empty
|
|
| modal_token_secret |
string
|
Modal token secret, if you don't know what this is, leave it empty
|
Output schema
The shape of the response you’ll get when you run this model with an API.
Schema
{'format': 'uri', 'title': 'Output', 'type': 'string'}