You're looking at a specific version of this model. Jump to the model overview.
phospho-app /gr00t-policy:6053b83c
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'
|
|
| wandb_api_key |
string
|
Weights & Biases API key (optional, to track the online training), find yours here: https://wandb.ai/authorize
|
|
| batch_size |
integer
|
16
Min: 1 Max: 32 |
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.0001
Min: 0.00001 Max: 0.01 |
Learning rate for training
|
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'}