LucaFrat/groot-bs128

GR00T fine-tune

Fine-tuned from nvidia/GR00T-N1.7-3B on 2026-05-27T00:28:30+00:00. Embodiment: UNITREE_G1_SONIC. Trained on 1 GPUs.

Hyperparameters

timestamp           = 2026-05-26T21:13:45+00:00
log_file            = /home/ubuntu/groot-files/logs/train-2026-05-26-211345.log
base_model          = nvidia/GR00T-N1.7-3B
embodiment_tag      = UNITREE_G1_SONIC
dataset_dir         = /home/ubuntu/groot-files/dataset
hf_dataset          = LucaFrat/dataset_100
checkpoint_dir      = /home/ubuntu/groot-files/checkpoints/g1_finetune-20260526-211344-gpu2
num_gpus            = 1
visible_gpu         = 2
global_batch_size   = 128
max_steps           = 20000
save_steps          = 4000
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-finetune
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.