training_config.json

{
  "freeze_transformer": false,
  "freeze_vision_encoder": true,
  "train_action_head_only": false,
  "optimizer_lr": 0.0001,
  "optimizer_betas": [
    0.9,
    0.999
  ],
  "optimizer_eps": 1e-08,
  "optimizer_weight_decay": 0.0001,
  "optimizer_grad_clip_norm": 10.0,
  "scheduler_warmup_steps": 1000,
  "scheduler_decay_steps": 100000,
  "scheduler_decay_lr": 1e-05,
  "push_to_hub": false
}