LucaFrat/groot-wbc-v2

GR00T fine-tune

Fine-tuned from nvidia/GR00T-N1.7-3B on 2026-05-28T08:36:41+00:00. Embodiment: UNITREE_G1_SONIC. Trained on 4 GPUs.

Hyperparameters

timestamp           = 2026-05-28T08:31:09+00:00
log_file            = /home/ubuntu/groot-files/logs/train-2026-05-28-083109.log
base_model          = nvidia/GR00T-N1.7-3B
embodiment_tag      = UNITREE_G1_SONIC
dataset_dir         = /home/ubuntu/groot-files/dataset_wbc_train
eval_dataset_dir    = /home/ubuntu/groot-files/dataset_wbc_eval
checkpoint_dir      = /home/ubuntu/groot-files/checkpoints/run-2026-05-28-083109
num_gpus            = 4
global_batch_size   = 32
max_steps           = 300
save_steps          = 50
save_total_limit    = 10
learning_rate       = 1e-4
warmup_ratio        = 0.05
weight_decay        = 1e-5
dataloader_workers  = 6
resume              = <none>
use_wandb           = 1
wandb_project       = groot-wbc
wandb_run_name      = <auto>

Checkpoints

Intermediate checkpoints (checkpoint-N/) are pushed incrementally during training; the top-level files in this repo are the final model state.