You're looking at a specific version of this model. Jump to the model overview.

phospho-app /gr00t-policy:3aacee05

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
hf_token
string
Hugging Face API token (used to upload your trained model to your HF profile), find yours here: https://huggingface.co/settings/token
hf_model_name
string
trained-gr00t-replicate
Hugging Face model name, the name of the model to be created, i.e. 'trained-act-replicate'
wandb_project
string
gr00t-Replicate
Weights & Biases project name
steps
integer
1

Min: 1

Max: 10000

Number of steps to train for

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'}