Add gr00t/80_20_gr00t (72 files)

gr00t/80_20_gr00t/checkpoint-10000/config.json

@@ -0,0 +1,70 @@
+{
+  "action_horizon": 50,
+  "add_pos_embed": true,
+  "apply_sincos_state_encoding": true,
+  "architectures": [
+    "Gr00tN1d6"
+  ],
+  "attn_dropout": 0.2,
+  "attn_implementation": null,
+  "backbone_embedding_dim": 2048,
+  "backbone_model_type": "eagle",
+  "backbone_trainable_params_fp32": true,
+  "collator_overwrite_image_inputs": false,
+  "color_jitter_params": {
+    "brightness": 0.1,
+    "contrast": 0.1,
+    "hue": 0.1,
+    "saturation": 0.1
+  },
+  "crop_fraction": 0.95,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "eagle_collator": true,
+  "formalize_language": true,
+  "gemma_collator": false,
+  "hidden_size": 1024,
+  "image_crop_size": null,
+  "image_target_size": null,
+  "input_embedding_dim": 1536,
+  "load_bf16": true,
+  "max_action_dim": 128,
+  "max_num_embodiments": 32,
+  "max_seq_len": 1024,
+  "max_state_dim": 128,
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "model_type": "Gr00tN1d6",
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_inference_timesteps": 4,
+  "num_timestep_buckets": 1000,
+  "random_rotation_angle": null,
+  "reproject_vision": false,
+  "select_layer": 16,
+  "shortest_image_edge": 256,
+  "state_dropout_prob": 0.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3",
+  "tune_diffusion_model": true,
+  "tune_llm": true,
+  "tune_projector": true,
+  "tune_top_llm_layers": 4,
+  "tune_visual": false,
+  "tune_vlln": true,
+  "use_albumentations_transforms": true,
+  "use_alternate_vl_dit": true,
+  "use_flash_attention": true,
+  "use_relative_action": true,
+  "use_vlln": true
+}

gr00t/80_20_gr00t/checkpoint-10000/embodiment_id.json

@@ -0,0 +1,11 @@
+{
+  "robocasa_panda_omron": 13,
+  "gr1": 20,
+  "behavior_r1_pro": 24,
+  "unitree_g1": 8,
+  "oxe_google": 0,
+  "oxe_widowx": 1,
+  "libero_panda": 2,
+  "oxe_droid": 16,
+  "new_embodiment": 10
+}

gr00t/80_20_gr00t/checkpoint-10000/experiment_cfg/conf.yaml

@@ -0,0 +1,246 @@
+load_config_path: null
+model:
+  model_type: Gr00tN1d6
+  model_dtype: bfloat16
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  backbone_model_type: eagle
+  model_revision: null
+  tune_top_llm_layers: 4
+  backbone_embedding_dim: 2048
+  tune_llm: true
+  tune_visual: false
+  select_layer: 16
+  reproject_vision: false
+  use_flash_attention: true
+  load_bf16: false
+  collator_overwrite_image_inputs: false
+  eagle_collator: true
+  backbone_trainable_params_fp32: true
+  image_crop_size: null
+  image_target_size: null
+  shortest_image_edge: 256
+  crop_fraction: 0.95
+  random_rotation_angle: null
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  use_albumentations_transforms: true
+  formalize_language: true
+  apply_sincos_state_encoding: false
+  use_relative_action: true
+  max_state_dim: 29
+  max_action_dim: 29
+  action_horizon: 16
+  hidden_size: 1024
+  input_embedding_dim: 1536
+  add_pos_embed: true
+  attn_dropout: 0.2
+  use_vlln: true
+  max_seq_len: 1024
+  use_alternate_vl_dit: true
+  attend_text_every_n_blocks: 2
+  diffusion_model_cfg:
+    positional_embeddings: null
+    num_layers: 32
+    num_attention_heads: 32
+    attention_head_dim: 48
+    norm_type: ada_norm
+    dropout: 0.2
+    final_dropout: true
+    output_dim: 1024
+    interleave_self_attention: true
+  num_inference_timesteps: 4
+  noise_beta_alpha: 1.5
+  noise_beta_beta: 1.0
+  noise_s: 0.999
+  num_timestep_buckets: 1000
+  tune_projector: true
+  tune_diffusion_model: true
+  tune_vlln: true
+  state_dropout_prob: 0.0
+  state_additive_noise_scale: 0.0
+  max_num_embodiments: 32
+data:
+  datasets:
+  - dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  modality_configs:
+    new_embodiment:
+      video:
+        delta_indices:
+        - 0
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      state:
+        delta_indices:
+        - 0
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      action:
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs:
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+      language:
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+  download_cache: false
+  shard_size: 1024
+  episode_sampling_rate: 0.1
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  mode: single_turn
+  random_chop: 0.0
+  mock_dataset_mode: false
+  shuffle: true
+  seed: 42
+  multiprocessing_context: fork
+  allow_padding: false
+  subsample_ratio: 1.0
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  video_backend: torchcodec
+training:
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  experiment_name: null
+  max_steps: 10000
+  global_batch_size: 32
+  batch_size: null
+  gradient_accumulation_steps: 1
+  learning_rate: 0.0001
+  lr_scheduler_type: cosine
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  max_grad_norm: 1.0
+  optim: adamw_torch
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  fp16: false
+  bf16: true
+  eval_bf16: true
+  logging_steps: 10
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  max_concurrent_uploads: 2
+  eval_strategy: 'no'
+  eval_steps: 500
+  eval_set_split_ratio: 0.1
+  eval_batch_size: 2
+  save_best_eval_metric_name: ''
+  save_best_eval_metric_greater_is_better: true
+  deepspeed_stage: 2
+  gradient_checkpointing: false
+  transformers_trust_remote_code: true
+  transformers_local_files_only: false
+  transformers_cache_dir: null
+  transformers_access_token: null
+  use_ddp: false
+  ddp_bucket_cap_mb: 100
+  num_gpus: 1
+  dataloader_num_workers: 4
+  remove_unused_columns: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  enable_profiling: false
+  max_retries: 3
+  assert_loss_less_than: null
+  add_rl_callback: false
+  enable_open_loop_eval: false
+  open_loop_eval_traj_ids:
+  - 0
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_plot_indices: null
+max_steps: 10000
+save_steps: 3000

gr00t/80_20_gr00t/checkpoint-10000/experiment_cfg/config.yaml

@@ -0,0 +1,277 @@
+!!python/object:gr00t.configs.base_config.Config
+data: !!python/object:gr00t.configs.data.data_config.DataConfig
+  allow_padding: false
+  datasets:
+  - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
+    dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    val_dataset_path: null
+  download_cache: false
+  episode_sampling_rate: 0.1
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  mock_dataset_mode: false
+  modality_configs:
+    new_embodiment:
+      action: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs:
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
+          - default
+          rep: &id002 !!python/object/apply:gr00t.data.types.ActionRepresentation
+          - absolute
+          state_key: null
+          type: &id003 !!python/object/apply:gr00t.data.types.ActionType
+          - non_eef
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        mean_std_embedding_keys: null
+        modality_keys: &id004
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+      language: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+      state: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys: *id004
+        sin_cos_embedding_keys: null
+      video: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+  mode: single_turn
+  multiprocessing_context: fork
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  random_chop: 0.0
+  seed: 42
+  shard_size: 1024
+  shuffle: true
+  subsample_ratio: 1.0
+  video_backend: torchcodec
+load_config_path: null
+model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
+  _attn_implementation_autoset: false
+  _attn_implementation_internal: null
+  _commit_hash: null
+  _name_or_path: ''
+  add_cross_attention: false
+  architectures: null
+  backbone_model_type: eagle
+  backbone_trainable_params_fp32: true
+  bad_words_ids: null
+  begin_suppress_tokens: null
+  bos_token_id: null
+  chunk_size_feed_forward: 0
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    hue: 0.08
+    saturation: 0.5
+  cross_attention_hidden_size: null
+  decoder_start_token_id: null
+  diffusion_model_cfg:
+    attention_head_dim: 48
+    dropout: 0.2
+    final_dropout: true
+    interleave_self_attention: true
+    norm_type: ada_norm
+    num_attention_heads: 32
+    num_layers: 32
+    output_dim: 1024
+    positional_embeddings: null
+  diversity_penalty: 0.0
+  do_sample: false
+  eagle_collator: true
+  early_stopping: false
+  encoder_no_repeat_ngram_size: 0
+  eos_token_id: null
+  exponential_decay_length_penalty: null
+  finetuning_task: null
+  forced_bos_token_id: null
+  forced_eos_token_id: null
+  id2label:
+    0: LABEL_0
+    1: LABEL_1
+  is_decoder: false
+  is_encoder_decoder: false
+  label2id:
+    LABEL_0: 0
+    LABEL_1: 1
+  length_penalty: 1.0
+  load_bf16: false
+  max_length: 20
+  min_length: 0
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  no_repeat_ngram_size: 0
+  num_beam_groups: 1
+  num_beams: 1
+  num_return_sequences: 1
+  output_attentions: false
+  output_hidden_states: false
+  output_scores: false
+  pad_token_id: null
+  prefix: null
+  problem_type: null
+  pruned_heads: {}
+  random_rotation_angle: null
+  remove_invalid_values: false
+  repetition_penalty: 1.0
+  reproject_vision: false
+  return_dict: true
+  return_dict_in_generate: false
+  sep_token_id: null
+  state_dropout_prob: 0.0
+  suppress_tokens: null
+  task_specific_params: null
+  temperature: 1.0
+  tf_legacy_loss: false
+  tie_encoder_decoder: false
+  tie_word_embeddings: true
+  tokenizer_class: null
+  top_k: 50
+  top_p: 1.0
+  torch_dtype: null
+  torchscript: false
+  transformers_version: null
+  tune_diffusion_model: true
+  tune_llm: true
+  tune_projector: true
+  tune_visual: false
+  typical_p: 1.0
+  use_bfloat16: false
+  use_relative_action: true
+training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
+  add_rl_callback: false
+  assert_loss_less_than: null
+  batch_size: null
+  bf16: true
+  dataloader_num_workers: 4
+  ddp_bucket_cap_mb: 100
+  deepspeed_stage: 2
+  enable_open_loop_eval: false
+  enable_profiling: false
+  eval_batch_size: 2
+  eval_bf16: true
+  eval_set_split_ratio: 0.1
+  eval_steps: 500
+  eval_strategy: 'no'
+  experiment_name: null
+  fp16: false
+  global_batch_size: 32
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: false
+  learning_rate: 0.0001
+  logging_steps: 10
+  lr_scheduler_type: cosine
+  max_concurrent_uploads: 2
+  max_grad_norm: 1.0
+  max_retries: 3
+  max_steps: 10000
+  num_gpus: 1
+  open_loop_eval_plot_indices: null
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_traj_ids:
+  - 0
+  optim: adamw_torch
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  transformers_access_token: null
+  transformers_cache_dir: null
+  transformers_local_files_only: false
+  transformers_trust_remote_code: true
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  use_ddp: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  weight_decay: 1.0e-05

gr00t/80_20_gr00t/checkpoint-10000/experiment_cfg/dataset_statistics.json

@@ -0,0 +1,761 @@
+{
+  "new_embodiment": {
+    "state": {
+      "left_arm": {
+        "min": [
+          -1.4764578342437744,
+          -0.1518283635377884,
+          -1.499962568283081,
+          -2.27626633644104,
+          -1.5671002864837646,
+          -0.6656755208969116,
+          -0.5207215547561646
+        ],
+        "max": [
+          0.3441833555698395,
+          0.657663881778717,
+          1.2758116722106934,
+          0.7662562131881714,
+          0.5798436999320984,
+          0.9854870438575745,
+          0.6866571307182312
+        ],
+        "mean": [
+          -0.24674749374389648,
+          0.27366140484809875,
+          -0.37971556186676025,
+          -1.63396155834198,
+          -0.1316976696252823,
+          0.2934497594833374,
+          -0.003799525322392583
+        ],
+        "std": [
+          0.2643950283527374,
+          0.06967416405677795,
+          0.5029074549674988,
+          0.685847282409668,
+          0.41868942975997925,
+          0.2417760193347931,
+          0.23273395001888275
+        ],
+        "q01": [
+          -1.1877377033233643,
+          0.13130172491073605,
+          -1.4007886266708374,
+          -2.166787986755371,
+          -1.5548944282531738,
+          -0.4001814353466034,
+          -0.34637925028800964
+        ],
+        "q99": [
+          0.10601000964642036,
+          0.46540972113609314,
+          1.1985569143295296,
+          0.6938967108726501,
+          0.2971799075603485,
+          0.9169320225715639,
+          0.5222405481338501
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7414746284484863,
+          -0.16363811492919922,
+          -0.741387128829956,
+          -0.6882719993591309,
+          -1.5692027807235718,
+          -1.5692027807235718
+        ],
+        "std": [
+          1.2994686365127563,
+          1.4938863515853882,
+          1.2995597124099731,
+          1.3310250043869019,
+          2.54817533493042,
+          2.54817533493042
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7119174599647522,
+          -0.55275958776474,
+          -0.09536845237016678,
+          -2.144679546356201,
+          -0.6652951240539551,
+          -0.7053864002227783,
+          -0.5874713063240051
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.433588981628418,
+          0.9927297830581665,
+          0.6599544286727905
+        ],
+        "mean": [
+          -0.1792619228363037,
+          -0.1758386492729187,
+          0.46474239230155945,
+          -1.5775669813156128,
+          0.06633679568767548,
+          -0.13522601127624512,
+          -0.0053453403525054455
+        ],
+        "std": [
+          0.2031964510679245,
+          0.17344926297664642,
+          0.2641066610813141,
+          0.44620281457901,
+          0.5178105235099792,
+          0.2680231034755707,
+          0.1985798329114914
+        ],
+        "q01": [
+          -0.5415496277809143,
+          -0.43221070885658264,
+          0.05797714069485664,
+          -2.007323226928711,
+          -0.5142281770706176,
+          -0.6404054927825927,
+          -0.3437119019031525
+        ],
+        "q99": [
+          0.2733417749404907,
+          0.4016014683246615,
+          1.2267546653747559,
+          0.6918689608573914,
+          1.3172417879104614,
+          0.8524364829063416,
+          0.4512056303024375
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40354546904563904,
+          0.11602523177862167,
+          -0.4034579396247864,
+          -0.4363398253917694,
+          -0.8933442831039429,
+          -0.8933442831039429
+        ],
+        "std": [
+          1.4348114728927612,
+          1.497140884399414,
+          1.434880018234253,
+          1.402381420135498,
+          2.8489983081817627,
+          2.8489983081817627
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "action": {
+      "left_arm": {
+        "min": [
+          -1.5419785976409912,
+          -0.19808706641197205,
+          -1.5707963705062866,
+          -2.3319737911224365,
+          -1.5707963705062866,
+          -0.6781747937202454,
+          -0.5912933349609375
+        ],
+        "max": [
+          0.339247465133667,
+          0.6958394646644592,
+          1.2758116722106934,
+          0.7663928866386414,
+          0.6010454297065735,
+          0.9854870438575745,
+          0.6981316804885864
+        ],
+        "mean": [
+          -0.27380800247192383,
+          0.27597638964653015,
+          -0.37349793314933777,
+          -1.6559933423995972,
+          -0.13736535608768463,
+          0.29294726252555847,
+          -0.005824246443808079
+        ],
+        "std": [
+          0.27345916628837585,
+          0.07336905598640442,
+          0.5032736659049988,
+          0.696763277053833,
+          0.4247623383998871,
+          0.24443820118904114,
+          0.23588497936725616
+        ],
+        "q01": [
+          -1.2448639106750488,
+          0.12081794798374176,
+          -1.4095649719238281,
+          -2.2054867362976074,
+          -1.5707963705062866,
+          -0.40510578513145445,
+          -0.34792898654937743
+        ],
+        "q99": [
+          0.08568856686353694,
+          0.4764096879959107,
+          1.1983782052993774,
+          0.6938967108726501,
+          0.3001057207584381,
+          0.9169320225715639,
+          0.5328076863288891
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7413919568061829,
+          -0.1634250432252884,
+          -0.7413044571876526,
+          -0.687811553478241,
+          -1.5692328214645386,
+          -1.5692328214645386
+        ],
+        "std": [
+          1.299504280090332,
+          1.4938620328903198,
+          1.2995966672897339,
+          1.3312314748764038,
+          2.5481374263763428,
+          2.5481374263763428
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7609559297561646,
+          -0.5897848010063171,
+          -0.13661164045333862,
+          -2.1815736293792725,
+          -0.6809128522872925,
+          -0.7053864002227783,
+          -0.6164834499359131
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.5061734914779663,
+          0.9927297830581665,
+          0.6635355353355408
+        ],
+        "mean": [
+          -0.20635083317756653,
+          -0.17740032076835632,
+          0.45279860496520996,
+          -1.6069952249526978,
+          0.0653856173157692,
+          -0.1357136219739914,
+          -0.009931313805282116
+        ],
+        "std": [
+          0.2129511535167694,
+          0.175509974360466,
+          0.26751890778541565,
+          0.45712384581565857,
+          0.5241751670837402,
+          0.2693447172641754,
+          0.20302622020244598
+        ],
+        "q01": [
+          -0.5830041313171387,
+          -0.44531570076942445,
+          0.04082439854741096,
+          -2.0516329193115235,
+          -0.5222506451606751,
+          -0.6403132677078247,
+          -0.3627131497859955
+        ],
+        "q99": [
+          0.2731454372406006,
+          0.4016782689094546,
+          1.2265138626098633,
+          0.6918689608573914,
+          1.3290673494338998,
+          0.8524364829063416,
+          0.45168524980545044
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40342941880226135,
+          0.1162116676568985,
+          -0.4033419191837311,
+          -0.43629777431488037,
+          -0.8933074474334717,
+          -0.8933074474334717
+        ],
+        "std": [
+          1.4348500967025757,
+          1.4971321821212769,
+          1.434924602508545,
+          1.4023455381393433,
+          2.849017381668091,
+          2.849017381668091
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "relative_action": {}
+  }
+}

gr00t/80_20_gr00t/checkpoint-10000/experiment_cfg/final_model_config.json

@@ -0,0 +1,53 @@
+{
+  "model_type": "Gr00tN1d6",
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "backbone_model_type": "eagle",
+  "model_revision": null,
+  "tune_top_llm_layers": 4,
+  "backbone_embedding_dim": 2048,
+  "tune_llm": true,
+  "tune_visual": false,
+  "select_layer": 16,
+  "reproject_vision": false,
+  "use_flash_attention": true,
+  "load_bf16": true,
+  "collator_overwrite_image_inputs": false,
+  "eagle_collator": true,
+  "backbone_trainable_params_fp32": true,
+  "apply_sincos_state_encoding": true,
+  "use_relative_action": true,
+  "max_state_dim": 128,
+  "max_action_dim": 128,
+  "action_horizon": 50,
+  "hidden_size": 1024,
+  "input_embedding_dim": 1536,
+  "add_pos_embed": true,
+  "attn_dropout": 0.2,
+  "use_vlln": true,
+  "max_seq_len": 1024,
+  "use_alternate_vl_dit": true,
+  "attend_text_every_n_blocks": 2,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "num_inference_timesteps": 4,
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_timestep_buckets": 1000,
+  "tune_projector": true,
+  "tune_diffusion_model": true,
+  "tune_vlln": true,
+  "state_dropout_prob": 0.0,
+  "state_additive_noise_scale": 0.0,
+  "max_num_embodiments": 32
+}

gr00t/80_20_gr00t/checkpoint-10000/model-00001-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00d648668a9a472f311e4f770f223320576b7630de9ac8949b930a05ced8add5
+size 4967164296

gr00t/80_20_gr00t/checkpoint-10000/model-00002-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6be9dd38debcf7626edb73a11035e2ba428d0730a95a94f0176168066e6238d8
+size 4982025552

gr00t/80_20_gr00t/checkpoint-10000/model-00003-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa58573b6a3b4c78d1a1f773da68d9158ead83536cade4e82011c0b5dd8fbe36
+size 2314751176

gr00t/80_20_gr00t/checkpoint-10000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0f95fee3843272794b0495bc96997bdddd58e63ea2eaeb1be8f7d5ae2cdf9f1
+size 20277711680

gr00t/80_20_gr00t/checkpoint-10000/processor_config.json

@@ -0,0 +1,504 @@
+{
+  "processor_class": "Gr00tN1d6Processor",
+  "processor_kwargs": {
+    "modality_configs": {
+      "behavior_r1_pro": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "observation.images.rgb.head_256_256",
+            "observation.images.rgb.left_wrist_256_256",
+            "observation.images.rgb.right_wrist_256_256"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "robot_pos",
+            "robot_ori_cos",
+            "robot_ori_sin",
+            "robot_2d_ori",
+            "robot_2d_ori_cos",
+            "robot_2d_ori_sin",
+            "robot_lin_vel",
+            "robot_ang_vel",
+            "arm_left_qpos",
+            "arm_left_qpos_sin",
+            "arm_left_qpos_cos",
+            "eef_left_pos",
+            "eef_left_quat",
+            "gripper_left_qpos",
+            "arm_right_qpos",
+            "arm_right_qpos_sin",
+            "arm_right_qpos_cos",
+            "eef_right_pos",
+            "eef_right_quat",
+            "gripper_right_qpos",
+            "trunk_qpos"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31
+          ],
+          "modality_keys": [
+            "base",
+            "torso",
+            "left_arm",
+            "left_gripper",
+            "right_arm",
+            "right_gripper"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "trunk_qpos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "arm_left_qpos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "arm_right_qpos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "gr1": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "ego_view_bg_crop_pad_res256_freq20"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "task"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "robocasa_panda_omron": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "res256_image_side_0",
+            "res256_image_side_1",
+            "res256_image_wrist_0"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "end_effector_position_relative",
+            "end_effector_rotation_relative",
+            "gripper_qpos",
+            "base_position",
+            "base_rotation"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "end_effector_position",
+            "end_effector_rotation",
+            "gripper_close",
+            "base_motion",
+            "control_mode"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.action.task_description"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "new_embodiment": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "ego_view",
+            "left_wrist_view",
+            "right_wrist_view"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_arm",
+            "left_hand",
+            "left_leg",
+            "neck",
+            "right_arm",
+            "right_hand",
+            "right_leg",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "left_arm",
+            "left_hand",
+            "left_leg",
+            "neck",
+            "right_arm",
+            "right_hand",
+            "right_leg",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.action.task_description"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      }
+    },
+    "image_crop_size": null,
+    "image_target_size": null,
+    "use_albumentations": true,
+    "random_rotation_angle": null,
+    "color_jitter_params": {
+      "brightness": 0.3,
+      "contrast": 0.4,
+      "saturation": 0.5,
+      "hue": 0.08
+    },
+    "shortest_image_edge": 256,
+    "crop_fraction": 0.95,
+    "model_name": "nvidia/Eagle-Block2A-2B-v2",
+    "model_type": "eagle",
+    "formalize_language": true,
+    "max_state_dim": 128,
+    "max_action_dim": 128,
+    "max_action_horizon": 50,
+    "use_percentiles": false,
+    "clip_outliers": true,
+    "apply_sincos_state_encoding": true,
+    "use_relative_action": true
+  }
+}

gr00t/80_20_gr00t/checkpoint-10000/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07c9c68f9f5a4ff7c2756642a3e4376e946cce84760795260e1e09012773add2
+size 14645

gr00t/80_20_gr00t/checkpoint-10000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d02be6d8bda4ea9c67040ed89f878acdc986bd4df3fbc60440a9d3eacca02d63
+size 1465

gr00t/80_20_gr00t/checkpoint-10000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d223ff78a7bd2f2dddd363e2f1515e00aaabad284080b6e91ca0664ba1755bd5
+size 5841

gr00t/80_20_gr00t/checkpoint-10000/wandb_config.json

@@ -0,0 +1 @@
+{"project": "finetune-gr00t-n1d6", "run_id": "80_20_gr00t"}

gr00t/80_20_gr00t/checkpoint-6000/config.json

@@ -0,0 +1,70 @@
+{
+  "action_horizon": 50,
+  "add_pos_embed": true,
+  "apply_sincos_state_encoding": true,
+  "architectures": [
+    "Gr00tN1d6"
+  ],
+  "attn_dropout": 0.2,
+  "attn_implementation": null,
+  "backbone_embedding_dim": 2048,
+  "backbone_model_type": "eagle",
+  "backbone_trainable_params_fp32": true,
+  "collator_overwrite_image_inputs": false,
+  "color_jitter_params": {
+    "brightness": 0.1,
+    "contrast": 0.1,
+    "hue": 0.1,
+    "saturation": 0.1
+  },
+  "crop_fraction": 0.95,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "eagle_collator": true,
+  "formalize_language": true,
+  "gemma_collator": false,
+  "hidden_size": 1024,
+  "image_crop_size": null,
+  "image_target_size": null,
+  "input_embedding_dim": 1536,
+  "load_bf16": true,
+  "max_action_dim": 128,
+  "max_num_embodiments": 32,
+  "max_seq_len": 1024,
+  "max_state_dim": 128,
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "model_type": "Gr00tN1d6",
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_inference_timesteps": 4,
+  "num_timestep_buckets": 1000,
+  "random_rotation_angle": null,
+  "reproject_vision": false,
+  "select_layer": 16,
+  "shortest_image_edge": 256,
+  "state_dropout_prob": 0.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3",
+  "tune_diffusion_model": true,
+  "tune_llm": true,
+  "tune_projector": true,
+  "tune_top_llm_layers": 4,
+  "tune_visual": false,
+  "tune_vlln": true,
+  "use_albumentations_transforms": true,
+  "use_alternate_vl_dit": true,
+  "use_flash_attention": true,
+  "use_relative_action": true,
+  "use_vlln": true
+}

gr00t/80_20_gr00t/checkpoint-6000/embodiment_id.json

@@ -0,0 +1,11 @@
+{
+  "robocasa_panda_omron": 13,
+  "gr1": 20,
+  "behavior_r1_pro": 24,
+  "unitree_g1": 8,
+  "oxe_google": 0,
+  "oxe_widowx": 1,
+  "libero_panda": 2,
+  "oxe_droid": 16,
+  "new_embodiment": 10
+}

gr00t/80_20_gr00t/checkpoint-6000/experiment_cfg/conf.yaml

@@ -0,0 +1,246 @@
+load_config_path: null
+model:
+  model_type: Gr00tN1d6
+  model_dtype: bfloat16
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  backbone_model_type: eagle
+  model_revision: null
+  tune_top_llm_layers: 4
+  backbone_embedding_dim: 2048
+  tune_llm: true
+  tune_visual: false
+  select_layer: 16
+  reproject_vision: false
+  use_flash_attention: true
+  load_bf16: false
+  collator_overwrite_image_inputs: false
+  eagle_collator: true
+  backbone_trainable_params_fp32: true
+  image_crop_size: null
+  image_target_size: null
+  shortest_image_edge: 256
+  crop_fraction: 0.95
+  random_rotation_angle: null
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  use_albumentations_transforms: true
+  formalize_language: true
+  apply_sincos_state_encoding: false
+  use_relative_action: true
+  max_state_dim: 29
+  max_action_dim: 29
+  action_horizon: 16
+  hidden_size: 1024
+  input_embedding_dim: 1536
+  add_pos_embed: true
+  attn_dropout: 0.2
+  use_vlln: true
+  max_seq_len: 1024
+  use_alternate_vl_dit: true
+  attend_text_every_n_blocks: 2
+  diffusion_model_cfg:
+    positional_embeddings: null
+    num_layers: 32
+    num_attention_heads: 32
+    attention_head_dim: 48
+    norm_type: ada_norm
+    dropout: 0.2
+    final_dropout: true
+    output_dim: 1024
+    interleave_self_attention: true
+  num_inference_timesteps: 4
+  noise_beta_alpha: 1.5
+  noise_beta_beta: 1.0
+  noise_s: 0.999
+  num_timestep_buckets: 1000
+  tune_projector: true
+  tune_diffusion_model: true
+  tune_vlln: true
+  state_dropout_prob: 0.0
+  state_additive_noise_scale: 0.0
+  max_num_embodiments: 32
+data:
+  datasets:
+  - dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  modality_configs:
+    new_embodiment:
+      video:
+        delta_indices:
+        - 0
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      state:
+        delta_indices:
+        - 0
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      action:
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs:
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+      language:
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+  download_cache: false
+  shard_size: 1024
+  episode_sampling_rate: 0.1
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  mode: single_turn
+  random_chop: 0.0
+  mock_dataset_mode: false
+  shuffle: true
+  seed: 42
+  multiprocessing_context: fork
+  allow_padding: false
+  subsample_ratio: 1.0
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  video_backend: torchcodec
+training:
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  experiment_name: null
+  max_steps: 10000
+  global_batch_size: 32
+  batch_size: null
+  gradient_accumulation_steps: 1
+  learning_rate: 0.0001
+  lr_scheduler_type: cosine
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  max_grad_norm: 1.0
+  optim: adamw_torch
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  fp16: false
+  bf16: true
+  eval_bf16: true
+  logging_steps: 10
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  max_concurrent_uploads: 2
+  eval_strategy: 'no'
+  eval_steps: 500
+  eval_set_split_ratio: 0.1
+  eval_batch_size: 2
+  save_best_eval_metric_name: ''
+  save_best_eval_metric_greater_is_better: true
+  deepspeed_stage: 2
+  gradient_checkpointing: false
+  transformers_trust_remote_code: true
+  transformers_local_files_only: false
+  transformers_cache_dir: null
+  transformers_access_token: null
+  use_ddp: false
+  ddp_bucket_cap_mb: 100
+  num_gpus: 1
+  dataloader_num_workers: 4
+  remove_unused_columns: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  enable_profiling: false
+  max_retries: 3
+  assert_loss_less_than: null
+  add_rl_callback: false
+  enable_open_loop_eval: false
+  open_loop_eval_traj_ids:
+  - 0
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_plot_indices: null
+max_steps: 10000
+save_steps: 3000

gr00t/80_20_gr00t/checkpoint-6000/experiment_cfg/config.yaml

@@ -0,0 +1,277 @@
+!!python/object:gr00t.configs.base_config.Config
+data: !!python/object:gr00t.configs.data.data_config.DataConfig
+  allow_padding: false
+  datasets:
+  - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
+    dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    val_dataset_path: null
+  download_cache: false
+  episode_sampling_rate: 0.1
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  mock_dataset_mode: false
+  modality_configs:
+    new_embodiment:
+      action: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs:
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
+          - default
+          rep: &id002 !!python/object/apply:gr00t.data.types.ActionRepresentation
+          - absolute
+          state_key: null
+          type: &id003 !!python/object/apply:gr00t.data.types.ActionType
+          - non_eef
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        mean_std_embedding_keys: null
+        modality_keys: &id004
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+      language: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+      state: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys: *id004
+        sin_cos_embedding_keys: null
+      video: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+  mode: single_turn
+  multiprocessing_context: fork
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  random_chop: 0.0
+  seed: 42
+  shard_size: 1024
+  shuffle: true
+  subsample_ratio: 1.0
+  video_backend: torchcodec
+load_config_path: null
+model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
+  _attn_implementation_autoset: false
+  _attn_implementation_internal: null
+  _commit_hash: null
+  _name_or_path: ''
+  add_cross_attention: false
+  architectures: null
+  backbone_model_type: eagle
+  backbone_trainable_params_fp32: true
+  bad_words_ids: null
+  begin_suppress_tokens: null
+  bos_token_id: null
+  chunk_size_feed_forward: 0
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    hue: 0.08
+    saturation: 0.5
+  cross_attention_hidden_size: null
+  decoder_start_token_id: null
+  diffusion_model_cfg:
+    attention_head_dim: 48
+    dropout: 0.2
+    final_dropout: true
+    interleave_self_attention: true
+    norm_type: ada_norm
+    num_attention_heads: 32
+    num_layers: 32
+    output_dim: 1024
+    positional_embeddings: null
+  diversity_penalty: 0.0
+  do_sample: false
+  eagle_collator: true
+  early_stopping: false
+  encoder_no_repeat_ngram_size: 0
+  eos_token_id: null
+  exponential_decay_length_penalty: null
+  finetuning_task: null
+  forced_bos_token_id: null
+  forced_eos_token_id: null
+  id2label:
+    0: LABEL_0
+    1: LABEL_1
+  is_decoder: false
+  is_encoder_decoder: false
+  label2id:
+    LABEL_0: 0
+    LABEL_1: 1
+  length_penalty: 1.0
+  load_bf16: false
+  max_length: 20
+  min_length: 0
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  no_repeat_ngram_size: 0
+  num_beam_groups: 1
+  num_beams: 1
+  num_return_sequences: 1
+  output_attentions: false
+  output_hidden_states: false
+  output_scores: false
+  pad_token_id: null
+  prefix: null
+  problem_type: null
+  pruned_heads: {}
+  random_rotation_angle: null
+  remove_invalid_values: false
+  repetition_penalty: 1.0
+  reproject_vision: false
+  return_dict: true
+  return_dict_in_generate: false
+  sep_token_id: null
+  state_dropout_prob: 0.0
+  suppress_tokens: null
+  task_specific_params: null
+  temperature: 1.0
+  tf_legacy_loss: false
+  tie_encoder_decoder: false
+  tie_word_embeddings: true
+  tokenizer_class: null
+  top_k: 50
+  top_p: 1.0
+  torch_dtype: null
+  torchscript: false
+  transformers_version: null
+  tune_diffusion_model: true
+  tune_llm: true
+  tune_projector: true
+  tune_visual: false
+  typical_p: 1.0
+  use_bfloat16: false
+  use_relative_action: true
+training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
+  add_rl_callback: false
+  assert_loss_less_than: null
+  batch_size: null
+  bf16: true
+  dataloader_num_workers: 4
+  ddp_bucket_cap_mb: 100
+  deepspeed_stage: 2
+  enable_open_loop_eval: false
+  enable_profiling: false
+  eval_batch_size: 2
+  eval_bf16: true
+  eval_set_split_ratio: 0.1
+  eval_steps: 500
+  eval_strategy: 'no'
+  experiment_name: null
+  fp16: false
+  global_batch_size: 32
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: false
+  learning_rate: 0.0001
+  logging_steps: 10
+  lr_scheduler_type: cosine
+  max_concurrent_uploads: 2
+  max_grad_norm: 1.0
+  max_retries: 3
+  max_steps: 10000
+  num_gpus: 1
+  open_loop_eval_plot_indices: null
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_traj_ids:
+  - 0
+  optim: adamw_torch
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  transformers_access_token: null
+  transformers_cache_dir: null
+  transformers_local_files_only: false
+  transformers_trust_remote_code: true
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  use_ddp: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  weight_decay: 1.0e-05

gr00t/80_20_gr00t/checkpoint-6000/experiment_cfg/dataset_statistics.json

@@ -0,0 +1,761 @@
+{
+  "new_embodiment": {
+    "state": {
+      "left_arm": {
+        "min": [
+          -1.4764578342437744,
+          -0.1518283635377884,
+          -1.499962568283081,
+          -2.27626633644104,
+          -1.5671002864837646,
+          -0.6656755208969116,
+          -0.5207215547561646
+        ],
+        "max": [
+          0.3441833555698395,
+          0.657663881778717,
+          1.2758116722106934,
+          0.7662562131881714,
+          0.5798436999320984,
+          0.9854870438575745,
+          0.6866571307182312
+        ],
+        "mean": [
+          -0.24674749374389648,
+          0.27366140484809875,
+          -0.37971556186676025,
+          -1.63396155834198,
+          -0.1316976696252823,
+          0.2934497594833374,
+          -0.003799525322392583
+        ],
+        "std": [
+          0.2643950283527374,
+          0.06967416405677795,
+          0.5029074549674988,
+          0.685847282409668,
+          0.41868942975997925,
+          0.2417760193347931,
+          0.23273395001888275
+        ],
+        "q01": [
+          -1.1877377033233643,
+          0.13130172491073605,
+          -1.4007886266708374,
+          -2.166787986755371,
+          -1.5548944282531738,
+          -0.4001814353466034,
+          -0.34637925028800964
+        ],
+        "q99": [
+          0.10601000964642036,
+          0.46540972113609314,
+          1.1985569143295296,
+          0.6938967108726501,
+          0.2971799075603485,
+          0.9169320225715639,
+          0.5222405481338501
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7414746284484863,
+          -0.16363811492919922,
+          -0.741387128829956,
+          -0.6882719993591309,
+          -1.5692027807235718,
+          -1.5692027807235718
+        ],
+        "std": [
+          1.2994686365127563,
+          1.4938863515853882,
+          1.2995597124099731,
+          1.3310250043869019,
+          2.54817533493042,
+          2.54817533493042
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7119174599647522,
+          -0.55275958776474,
+          -0.09536845237016678,
+          -2.144679546356201,
+          -0.6652951240539551,
+          -0.7053864002227783,
+          -0.5874713063240051
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.433588981628418,
+          0.9927297830581665,
+          0.6599544286727905
+        ],
+        "mean": [
+          -0.1792619228363037,
+          -0.1758386492729187,
+          0.46474239230155945,
+          -1.5775669813156128,
+          0.06633679568767548,
+          -0.13522601127624512,
+          -0.0053453403525054455
+        ],
+        "std": [
+          0.2031964510679245,
+          0.17344926297664642,
+          0.2641066610813141,
+          0.44620281457901,
+          0.5178105235099792,
+          0.2680231034755707,
+          0.1985798329114914
+        ],
+        "q01": [
+          -0.5415496277809143,
+          -0.43221070885658264,
+          0.05797714069485664,
+          -2.007323226928711,
+          -0.5142281770706176,
+          -0.6404054927825927,
+          -0.3437119019031525
+        ],
+        "q99": [
+          0.2733417749404907,
+          0.4016014683246615,
+          1.2267546653747559,
+          0.6918689608573914,
+          1.3172417879104614,
+          0.8524364829063416,
+          0.4512056303024375
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40354546904563904,
+          0.11602523177862167,
+          -0.4034579396247864,
+          -0.4363398253917694,
+          -0.8933442831039429,
+          -0.8933442831039429
+        ],
+        "std": [
+          1.4348114728927612,
+          1.497140884399414,
+          1.434880018234253,
+          1.402381420135498,
+          2.8489983081817627,
+          2.8489983081817627
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "action": {
+      "left_arm": {
+        "min": [
+          -1.5419785976409912,
+          -0.19808706641197205,
+          -1.5707963705062866,
+          -2.3319737911224365,
+          -1.5707963705062866,
+          -0.6781747937202454,
+          -0.5912933349609375
+        ],
+        "max": [
+          0.339247465133667,
+          0.6958394646644592,
+          1.2758116722106934,
+          0.7663928866386414,
+          0.6010454297065735,
+          0.9854870438575745,
+          0.6981316804885864
+        ],
+        "mean": [
+          -0.27380800247192383,
+          0.27597638964653015,
+          -0.37349793314933777,
+          -1.6559933423995972,
+          -0.13736535608768463,
+          0.29294726252555847,
+          -0.005824246443808079
+        ],
+        "std": [
+          0.27345916628837585,
+          0.07336905598640442,
+          0.5032736659049988,
+          0.696763277053833,
+          0.4247623383998871,
+          0.24443820118904114,
+          0.23588497936725616
+        ],
+        "q01": [
+          -1.2448639106750488,
+          0.12081794798374176,
+          -1.4095649719238281,
+          -2.2054867362976074,
+          -1.5707963705062866,
+          -0.40510578513145445,
+          -0.34792898654937743
+        ],
+        "q99": [
+          0.08568856686353694,
+          0.4764096879959107,
+          1.1983782052993774,
+          0.6938967108726501,
+          0.3001057207584381,
+          0.9169320225715639,
+          0.5328076863288891
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7413919568061829,
+          -0.1634250432252884,
+          -0.7413044571876526,
+          -0.687811553478241,
+          -1.5692328214645386,
+          -1.5692328214645386
+        ],
+        "std": [
+          1.299504280090332,
+          1.4938620328903198,
+          1.2995966672897339,
+          1.3312314748764038,
+          2.5481374263763428,
+          2.5481374263763428
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7609559297561646,
+          -0.5897848010063171,
+          -0.13661164045333862,
+          -2.1815736293792725,
+          -0.6809128522872925,
+          -0.7053864002227783,
+          -0.6164834499359131
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.5061734914779663,
+          0.9927297830581665,
+          0.6635355353355408
+        ],
+        "mean": [
+          -0.20635083317756653,
+          -0.17740032076835632,
+          0.45279860496520996,
+          -1.6069952249526978,
+          0.0653856173157692,
+          -0.1357136219739914,
+          -0.009931313805282116
+        ],
+        "std": [
+          0.2129511535167694,
+          0.175509974360466,
+          0.26751890778541565,
+          0.45712384581565857,
+          0.5241751670837402,
+          0.2693447172641754,
+          0.20302622020244598
+        ],
+        "q01": [
+          -0.5830041313171387,
+          -0.44531570076942445,
+          0.04082439854741096,
+          -2.0516329193115235,
+          -0.5222506451606751,
+          -0.6403132677078247,
+          -0.3627131497859955
+        ],
+        "q99": [
+          0.2731454372406006,
+          0.4016782689094546,
+          1.2265138626098633,
+          0.6918689608573914,
+          1.3290673494338998,
+          0.8524364829063416,
+          0.45168524980545044
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40342941880226135,
+          0.1162116676568985,
+          -0.4033419191837311,
+          -0.43629777431488037,
+          -0.8933074474334717,
+          -0.8933074474334717
+        ],
+        "std": [
+          1.4348500967025757,
+          1.4971321821212769,
+          1.434924602508545,
+          1.4023455381393433,
+          2.849017381668091,
+          2.849017381668091
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "relative_action": {}
+  }
+}

gr00t/80_20_gr00t/checkpoint-6000/experiment_cfg/final_model_config.json

@@ -0,0 +1,53 @@
+{
+  "model_type": "Gr00tN1d6",
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "backbone_model_type": "eagle",
+  "model_revision": null,
+  "tune_top_llm_layers": 4,
+  "backbone_embedding_dim": 2048,
+  "tune_llm": true,
+  "tune_visual": false,
+  "select_layer": 16,
+  "reproject_vision": false,
+  "use_flash_attention": true,
+  "load_bf16": true,
+  "collator_overwrite_image_inputs": false,
+  "eagle_collator": true,
+  "backbone_trainable_params_fp32": true,
+  "apply_sincos_state_encoding": true,
+  "use_relative_action": true,
+  "max_state_dim": 128,
+  "max_action_dim": 128,
+  "action_horizon": 50,
+  "hidden_size": 1024,
+  "input_embedding_dim": 1536,
+  "add_pos_embed": true,
+  "attn_dropout": 0.2,
+  "use_vlln": true,
+  "max_seq_len": 1024,
+  "use_alternate_vl_dit": true,
+  "attend_text_every_n_blocks": 2,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "num_inference_timesteps": 4,
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_timestep_buckets": 1000,
+  "tune_projector": true,
+  "tune_diffusion_model": true,
+  "tune_vlln": true,
+  "state_dropout_prob": 0.0,
+  "state_additive_noise_scale": 0.0,
+  "max_num_embodiments": 32
+}

gr00t/80_20_gr00t/checkpoint-6000/model-00001-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fe695246bf949815b138fcbd9aaeef0fccbe2c8e4503f3ff58e232887154661
+size 4967164296

gr00t/80_20_gr00t/checkpoint-6000/model-00002-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db2c62aae28f7a05314b19d5b44288b04043ec4aab2834d212a260d7eef2d62c
+size 4982025552

gr00t/80_20_gr00t/checkpoint-6000/model-00003-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f24fa180cd52ad7d5798d7a802ff5fa8877aefc6ab5fc984e48e7d44892cec6b
+size 2314751176

gr00t/80_20_gr00t/checkpoint-6000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65b3598b79367dd97f69a8af664ce2242d7fda264b356df3ba4de5ecdb426b02
+size 20277711680

gr00t/80_20_gr00t/checkpoint-6000/processor_config.json

@@ -0,0 +1,504 @@
+{
+  "processor_class": "Gr00tN1d6Processor",
+  "processor_kwargs": {
+    "modality_configs": {
+      "behavior_r1_pro": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "observation.images.rgb.head_256_256",
+            "observation.images.rgb.left_wrist_256_256",
+            "observation.images.rgb.right_wrist_256_256"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "robot_pos",
+            "robot_ori_cos",
+            "robot_ori_sin",
+            "robot_2d_ori",
+            "robot_2d_ori_cos",
+            "robot_2d_ori_sin",
+            "robot_lin_vel",
+            "robot_ang_vel",
+            "arm_left_qpos",
+            "arm_left_qpos_sin",
+            "arm_left_qpos_cos",
+            "eef_left_pos",
+            "eef_left_quat",
+            "gripper_left_qpos",
+            "arm_right_qpos",
+            "arm_right_qpos_sin",
+            "arm_right_qpos_cos",
+            "eef_right_pos",
+            "eef_right_quat",
+            "gripper_right_qpos",
+            "trunk_qpos"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31
+          ],
+          "modality_keys": [
+            "base",
+            "torso",
+            "left_arm",
+            "left_gripper",
+            "right_arm",
+            "right_gripper"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "trunk_qpos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "arm_left_qpos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "arm_right_qpos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "gr1": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "ego_view_bg_crop_pad_res256_freq20"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "left_arm",
+            "right_arm",
+            "left_hand",
+            "right_hand",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "task"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "robocasa_panda_omron": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "res256_image_side_0",
+            "res256_image_side_1",
+            "res256_image_wrist_0"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "end_effector_position_relative",
+            "end_effector_rotation_relative",
+            "gripper_qpos",
+            "base_position",
+            "base_rotation"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "end_effector_position",
+            "end_effector_rotation",
+            "gripper_close",
+            "base_motion",
+            "control_mode"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.action.task_description"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "new_embodiment": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "ego_view",
+            "left_wrist_view",
+            "right_wrist_view"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_arm",
+            "left_hand",
+            "left_leg",
+            "neck",
+            "right_arm",
+            "right_hand",
+            "right_leg",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "action": {
+          "delta_indices": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15
+          ],
+          "modality_keys": [
+            "left_arm",
+            "left_hand",
+            "left_leg",
+            "neck",
+            "right_arm",
+            "right_hand",
+            "right_leg",
+            "waist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": null
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.action.task_description"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      }
+    },
+    "image_crop_size": null,
+    "image_target_size": null,
+    "use_albumentations": true,
+    "random_rotation_angle": null,
+    "color_jitter_params": {
+      "brightness": 0.3,
+      "contrast": 0.4,
+      "saturation": 0.5,
+      "hue": 0.08
+    },
+    "shortest_image_edge": 256,
+    "crop_fraction": 0.95,
+    "model_name": "nvidia/Eagle-Block2A-2B-v2",
+    "model_type": "eagle",
+    "formalize_language": true,
+    "max_state_dim": 128,
+    "max_action_dim": 128,
+    "max_action_horizon": 50,
+    "use_percentiles": false,
+    "clip_outliers": true,
+    "apply_sincos_state_encoding": true,
+    "use_relative_action": true
+  }
+}

gr00t/80_20_gr00t/checkpoint-6000/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f476c85815803955f3f0b961be1cdb4439238981e60db7010f58e8353145785
+size 14645

gr00t/80_20_gr00t/checkpoint-6000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c81168c968205dd4f18f99afb8d250e3039fa2276a497474189eec84cb61b541
+size 1465

gr00t/80_20_gr00t/checkpoint-6000/trainer_state.json

@@ -0,0 +1,3634 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.6,
+  "eval_steps": 500,
+  "global_step": 6000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 1.5147156715393066,
+      "learning_rate": 1.8e-06,
+      "loss": 1.3865,
+      "step": 10
+    },
+    {
+      "grad_norm": 1.0042343139648438,
+      "learning_rate": 3.8e-06,
+      "loss": 1.3706,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.8428165316581726,
+      "learning_rate": 5.8e-06,
+      "loss": 1.3381,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.7965219616889954,
+      "learning_rate": 7.8e-06,
+      "loss": 1.29,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.8637328743934631,
+      "learning_rate": 9.800000000000001e-06,
+      "loss": 1.236,
+      "step": 50
+    },
+    {
+      "grad_norm": 1.2560558319091797,
+      "learning_rate": 1.18e-05,
+      "loss": 1.1825,
+      "step": 60
+    },
+    {
+      "grad_norm": 1.4382917881011963,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 1.1241,
+      "step": 70
+    },
+    {
+      "grad_norm": 3.0528624057769775,
+      "learning_rate": 1.58e-05,
+      "loss": 1.0973,
+      "step": 80
+    },
+    {
+      "grad_norm": 2.5564255714416504,
+      "learning_rate": 1.78e-05,
+      "loss": 1.0839,
+      "step": 90
+    },
+    {
+      "grad_norm": 1.837116003036499,
+      "learning_rate": 1.9800000000000004e-05,
+      "loss": 1.0722,
+      "step": 100
+    },
+    {
+      "grad_norm": 1.3364044427871704,
+      "learning_rate": 2.18e-05,
+      "loss": 1.0701,
+      "step": 110
+    },
+    {
+      "grad_norm": 1.4615561962127686,
+      "learning_rate": 2.38e-05,
+      "loss": 1.056,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.564803957939148,
+      "learning_rate": 2.58e-05,
+      "loss": 1.0428,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.5108850598335266,
+      "learning_rate": 2.7800000000000005e-05,
+      "loss": 1.0443,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.4132338762283325,
+      "learning_rate": 2.98e-05,
+      "loss": 1.0302,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.7859615683555603,
+      "learning_rate": 3.18e-05,
+      "loss": 1.0329,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.3863542973995209,
+      "learning_rate": 3.38e-05,
+      "loss": 1.0317,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.40063974261283875,
+      "learning_rate": 3.58e-05,
+      "loss": 1.0338,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.5341059565544128,
+      "learning_rate": 3.7800000000000004e-05,
+      "loss": 1.0261,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.4857071042060852,
+      "learning_rate": 3.9800000000000005e-05,
+      "loss": 1.0312,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.4560174345970154,
+      "learning_rate": 4.18e-05,
+      "loss": 1.0231,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.5137712359428406,
+      "learning_rate": 4.38e-05,
+      "loss": 1.0039,
+      "step": 220
+    },
+    {
+      "grad_norm": 0.6078010201454163,
+      "learning_rate": 4.58e-05,
+      "loss": 0.999,
+      "step": 230
+    },
+    {
+      "grad_norm": 0.6455978155136108,
+      "learning_rate": 4.78e-05,
+      "loss": 0.9798,
+      "step": 240
+    },
+    {
+      "grad_norm": 0.6929338574409485,
+      "learning_rate": 4.9800000000000004e-05,
+      "loss": 0.9528,
+      "step": 250
+    },
+    {
+      "grad_norm": 1.4602608680725098,
+      "learning_rate": 5.1800000000000005e-05,
+      "loss": 0.9278,
+      "step": 260
+    },
+    {
+      "grad_norm": 3.2193353176116943,
+      "learning_rate": 5.380000000000001e-05,
+      "loss": 0.8963,
+      "step": 270
+    },
+    {
+      "grad_norm": 0.8640987277030945,
+      "learning_rate": 5.580000000000001e-05,
+      "loss": 0.8561,
+      "step": 280
+    },
+    {
+      "grad_norm": 0.6915462613105774,
+      "learning_rate": 5.7799999999999995e-05,
+      "loss": 0.8233,
+      "step": 290
+    },
+    {
+      "grad_norm": 0.9272186160087585,
+      "learning_rate": 5.9800000000000003e-05,
+      "loss": 0.771,
+      "step": 300
+    },
+    {
+      "grad_norm": 0.965694010257721,
+      "learning_rate": 6.18e-05,
+      "loss": 0.7411,
+      "step": 310
+    },
+    {
+      "grad_norm": 0.804237961769104,
+      "learning_rate": 6.38e-05,
+      "loss": 0.7042,
+      "step": 320
+    },
+    {
+      "grad_norm": 0.9388341903686523,
+      "learning_rate": 6.58e-05,
+      "loss": 0.6753,
+      "step": 330
+    },
+    {
+      "grad_norm": 0.7586079239845276,
+      "learning_rate": 6.780000000000001e-05,
+      "loss": 0.6428,
+      "step": 340
+    },
+    {
+      "grad_norm": 0.8676110506057739,
+      "learning_rate": 6.98e-05,
+      "loss": 0.6185,
+      "step": 350
+    },
+    {
+      "grad_norm": 0.7896904349327087,
+      "learning_rate": 7.18e-05,
+      "loss": 0.6032,
+      "step": 360
+    },
+    {
+      "grad_norm": 0.9266006350517273,
+      "learning_rate": 7.38e-05,
+      "loss": 0.5896,
+      "step": 370
+    },
+    {
+      "grad_norm": 0.8928837180137634,
+      "learning_rate": 7.58e-05,
+      "loss": 0.5605,
+      "step": 380
+    },
+    {
+      "grad_norm": 0.8889243602752686,
+      "learning_rate": 7.780000000000001e-05,
+      "loss": 0.5477,
+      "step": 390
+    },
+    {
+      "grad_norm": 0.7756143808364868,
+      "learning_rate": 7.98e-05,
+      "loss": 0.5203,
+      "step": 400
+    },
+    {
+      "grad_norm": 0.7256381511688232,
+      "learning_rate": 8.18e-05,
+      "loss": 0.5083,
+      "step": 410
+    },
+    {
+      "grad_norm": 0.9079662561416626,
+      "learning_rate": 8.38e-05,
+      "loss": 0.4887,
+      "step": 420
+    },
+    {
+      "grad_norm": 0.9908015131950378,
+      "learning_rate": 8.58e-05,
+      "loss": 0.4895,
+      "step": 430
+    },
+    {
+      "grad_norm": 0.943317711353302,
+      "learning_rate": 8.78e-05,
+      "loss": 0.4813,
+      "step": 440
+    },
+    {
+      "grad_norm": 0.8873269557952881,
+      "learning_rate": 8.98e-05,
+      "loss": 0.4752,
+      "step": 450
+    },
+    {
+      "grad_norm": 0.8335803151130676,
+      "learning_rate": 9.180000000000001e-05,
+      "loss": 0.4669,
+      "step": 460
+    },
+    {
+      "grad_norm": 0.7145240306854248,
+      "learning_rate": 9.38e-05,
+      "loss": 0.4607,
+      "step": 470
+    },
+    {
+      "grad_norm": 0.8512395024299622,
+      "learning_rate": 9.58e-05,
+      "loss": 0.4508,
+      "step": 480
+    },
+    {
+      "grad_norm": 0.7428747415542603,
+      "learning_rate": 9.78e-05,
+      "loss": 0.4407,
+      "step": 490
+    },
+    {
+      "grad_norm": 0.780677080154419,
+      "learning_rate": 9.98e-05,
+      "loss": 0.43,
+      "step": 500
+    },
+    {
+      "grad_norm": 0.6897915005683899,
+      "learning_rate": 9.9999778549206e-05,
+      "loss": 0.4253,
+      "step": 510
+    },
+    {
+      "grad_norm": 0.7703121900558472,
+      "learning_rate": 9.999901304280685e-05,
+      "loss": 0.4133,
+      "step": 520
+    },
+    {
+      "grad_norm": 0.7515524625778198,
+      "learning_rate": 9.999770075521164e-05,
+      "loss": 0.4077,
+      "step": 530
+    },
+    {
+      "grad_norm": 0.8542043566703796,
+      "learning_rate": 9.99958417007713e-05,
+      "loss": 0.3917,
+      "step": 540
+    },
+    {
+      "grad_norm": 0.9805722236633301,
+      "learning_rate": 9.999343589981615e-05,
+      "loss": 0.3786,
+      "step": 550
+    },
+    {
+      "grad_norm": 0.8073768615722656,
+      "learning_rate": 9.999048337865568e-05,
+      "loss": 0.3658,
+      "step": 560
+    },
+    {
+      "grad_norm": 0.8363681435585022,
+      "learning_rate": 9.998698416957815e-05,
+      "loss": 0.3642,
+      "step": 570
+    },
+    {
+      "grad_norm": 0.6901201009750366,
+      "learning_rate": 9.998293831085037e-05,
+      "loss": 0.3577,
+      "step": 580
+    },
+    {
+      "grad_norm": 0.7889610528945923,
+      "learning_rate": 9.997834584671719e-05,
+      "loss": 0.3574,
+      "step": 590
+    },
+    {
+      "grad_norm": 0.6903965473175049,
+      "learning_rate": 9.997320682740107e-05,
+      "loss": 0.355,
+      "step": 600
+    },
+    {
+      "grad_norm": 0.7509908080101013,
+      "learning_rate": 9.996752130910149e-05,
+      "loss": 0.3494,
+      "step": 610
+    },
+    {
+      "grad_norm": 0.61469966173172,
+      "learning_rate": 9.99612893539944e-05,
+      "loss": 0.3348,
+      "step": 620
+    },
+    {
+      "grad_norm": 0.6821316480636597,
+      "learning_rate": 9.995451103023144e-05,
+      "loss": 0.3294,
+      "step": 630
+    },
+    {
+      "grad_norm": 0.8215420246124268,
+      "learning_rate": 9.994718641193928e-05,
+      "loss": 0.331,
+      "step": 640
+    },
+    {
+      "grad_norm": 0.7098703384399414,
+      "learning_rate": 9.993931557921874e-05,
+      "loss": 0.328,
+      "step": 650
+    },
+    {
+      "grad_norm": 0.7727073431015015,
+      "learning_rate": 9.993089861814402e-05,
+      "loss": 0.3291,
+      "step": 660
+    },
+    {
+      "grad_norm": 0.8537644743919373,
+      "learning_rate": 9.992193562076166e-05,
+      "loss": 0.3405,
+      "step": 670
+    },
+    {
+      "grad_norm": 0.8080107569694519,
+      "learning_rate": 9.991242668508954e-05,
+      "loss": 0.3212,
+      "step": 680
+    },
+    {
+      "grad_norm": 0.7913404107093811,
+      "learning_rate": 9.990237191511587e-05,
+      "loss": 0.3232,
+      "step": 690
+    },
+    {
+      "grad_norm": 0.6615907549858093,
+      "learning_rate": 9.989177142079802e-05,
+      "loss": 0.3191,
+      "step": 700
+    },
+    {
+      "grad_norm": 0.683260977268219,
+      "learning_rate": 9.988062531806126e-05,
+      "loss": 0.3212,
+      "step": 710
+    },
+    {
+      "grad_norm": 0.6291384100914001,
+      "learning_rate": 9.986893372879762e-05,
+      "loss": 0.3061,
+      "step": 720
+    },
+    {
+      "grad_norm": 0.7944923043251038,
+      "learning_rate": 9.985669678086443e-05,
+      "loss": 0.3026,
+      "step": 730
+    },
+    {
+      "grad_norm": 0.6704161167144775,
+      "learning_rate": 9.984391460808298e-05,
+      "loss": 0.2872,
+      "step": 740
+    },
+    {
+      "grad_norm": 0.6949629187583923,
+      "learning_rate": 9.983058735023709e-05,
+      "loss": 0.288,
+      "step": 750
+    },
+    {
+      "grad_norm": 0.7024264335632324,
+      "learning_rate": 9.98167151530715e-05,
+      "loss": 0.2818,
+      "step": 760
+    },
+    {
+      "grad_norm": 0.6466235518455505,
+      "learning_rate": 9.980229816829034e-05,
+      "loss": 0.2815,
+      "step": 770
+    },
+    {
+      "grad_norm": 0.8368349075317383,
+      "learning_rate": 9.978733655355544e-05,
+      "loss": 0.2884,
+      "step": 780
+    },
+    {
+      "grad_norm": 0.9108396768569946,
+      "learning_rate": 9.977183047248464e-05,
+      "loss": 0.2895,
+      "step": 790
+    },
+    {
+      "grad_norm": 0.703660249710083,
+      "learning_rate": 9.975578009464992e-05,
+      "loss": 0.2917,
+      "step": 800
+    },
+    {
+      "grad_norm": 0.6723619699478149,
+      "learning_rate": 9.97391855955757e-05,
+      "loss": 0.2876,
+      "step": 810
+    },
+    {
+      "grad_norm": 0.6325339674949646,
+      "learning_rate": 9.972204715673669e-05,
+      "loss": 0.2795,
+      "step": 820
+    },
+    {
+      "grad_norm": 0.7178779244422913,
+      "learning_rate": 9.970436496555617e-05,
+      "loss": 0.2748,
+      "step": 830
+    },
+    {
+      "grad_norm": 0.6111972332000732,
+      "learning_rate": 9.968613921540373e-05,
+      "loss": 0.2749,
+      "step": 840
+    },
+    {
+      "grad_norm": 0.5450002551078796,
+      "learning_rate": 9.966737010559326e-05,
+      "loss": 0.2816,
+      "step": 850
+    },
+    {
+      "grad_norm": 0.7340812087059021,
+      "learning_rate": 9.964805784138072e-05,
+      "loss": 0.2726,
+      "step": 860
+    },
+    {
+      "grad_norm": 0.6444380283355713,
+      "learning_rate": 9.962820263396195e-05,
+      "loss": 0.2744,
+      "step": 870
+    },
+    {
+      "grad_norm": 0.6659711003303528,
+      "learning_rate": 9.960780470047033e-05,
+      "loss": 0.268,
+      "step": 880
+    },
+    {
+      "grad_norm": 0.8926224708557129,
+      "learning_rate": 9.958686426397437e-05,
+      "loss": 0.27,
+      "step": 890
+    },
+    {
+      "grad_norm": 0.5907139778137207,
+      "learning_rate": 9.956538155347534e-05,
+      "loss": 0.2788,
+      "step": 900
+    },
+    {
+      "grad_norm": 0.6686886548995972,
+      "learning_rate": 9.95433568039047e-05,
+      "loss": 0.262,
+      "step": 910
+    },
+    {
+      "grad_norm": 0.942680835723877,
+      "learning_rate": 9.952079025612162e-05,
+      "loss": 0.2639,
+      "step": 920
+    },
+    {
+      "grad_norm": 0.7353556752204895,
+      "learning_rate": 9.949768215691022e-05,
+      "loss": 0.2504,
+      "step": 930
+    },
+    {
+      "grad_norm": 0.8252507448196411,
+      "learning_rate": 9.9474032758977e-05,
+      "loss": 0.2535,
+      "step": 940
+    },
+    {
+      "grad_norm": 0.8672978281974792,
+      "learning_rate": 9.944984232094794e-05,
+      "loss": 0.2529,
+      "step": 950
+    },
+    {
+      "grad_norm": 0.6793451905250549,
+      "learning_rate": 9.942511110736584e-05,
+      "loss": 0.2388,
+      "step": 960
+    },
+    {
+      "grad_norm": 0.7761439085006714,
+      "learning_rate": 9.939983938868726e-05,
+      "loss": 0.2348,
+      "step": 970
+    },
+    {
+      "grad_norm": 0.7596891522407532,
+      "learning_rate": 9.93740274412797e-05,
+      "loss": 0.244,
+      "step": 980
+    },
+    {
+      "grad_norm": 0.6909441947937012,
+      "learning_rate": 9.934767554741846e-05,
+      "loss": 0.2392,
+      "step": 990
+    },
+    {
+      "grad_norm": 0.6320341229438782,
+      "learning_rate": 9.932078399528361e-05,
+      "loss": 0.2304,
+      "step": 1000
+    },
+    {
+      "grad_norm": 0.7369469404220581,
+      "learning_rate": 9.929335307895689e-05,
+      "loss": 0.2283,
+      "step": 1010
+    },
+    {
+      "grad_norm": 0.5903791189193726,
+      "learning_rate": 9.926538309841839e-05,
+      "loss": 0.2266,
+      "step": 1020
+    },
+    {
+      "grad_norm": 0.537055492401123,
+      "learning_rate": 9.923687435954334e-05,
+      "loss": 0.231,
+      "step": 1030
+    },
+    {
+      "grad_norm": 0.7914230823516846,
+      "learning_rate": 9.920782717409873e-05,
+      "loss": 0.2287,
+      "step": 1040
+    },
+    {
+      "grad_norm": 0.77292799949646,
+      "learning_rate": 9.917824185973994e-05,
+      "loss": 0.224,
+      "step": 1050
+    },
+    {
+      "grad_norm": 0.628867506980896,
+      "learning_rate": 9.914811874000723e-05,
+      "loss": 0.239,
+      "step": 1060
+    },
+    {
+      "grad_norm": 0.7403140664100647,
+      "learning_rate": 9.911745814432218e-05,
+      "loss": 0.2271,
+      "step": 1070
+    },
+    {
+      "grad_norm": 0.7243335247039795,
+      "learning_rate": 9.90862604079842e-05,
+      "loss": 0.2248,
+      "step": 1080
+    },
+    {
+      "grad_norm": 0.6691111922264099,
+      "learning_rate": 9.90545258721667e-05,
+      "loss": 0.2181,
+      "step": 1090
+    },
+    {
+      "grad_norm": 0.7756068706512451,
+      "learning_rate": 9.90222548839135e-05,
+      "loss": 0.216,
+      "step": 1100
+    },
+    {
+      "grad_norm": 0.7853686213493347,
+      "learning_rate": 9.898944779613495e-05,
+      "loss": 0.2212,
+      "step": 1110
+    },
+    {
+      "grad_norm": 0.6170802712440491,
+      "learning_rate": 9.89561049676041e-05,
+      "loss": 0.2112,
+      "step": 1120
+    },
+    {
+      "grad_norm": 0.6513901352882385,
+      "learning_rate": 9.89222267629528e-05,
+      "loss": 0.2148,
+      "step": 1130
+    },
+    {
+      "grad_norm": 0.6521427035331726,
+      "learning_rate": 9.888781355266763e-05,
+      "loss": 0.2091,
+      "step": 1140
+    },
+    {
+      "grad_norm": 0.5923445820808411,
+      "learning_rate": 9.885286571308598e-05,
+      "loss": 0.2097,
+      "step": 1150
+    },
+    {
+      "grad_norm": 0.6281879544258118,
+      "learning_rate": 9.881738362639182e-05,
+      "loss": 0.2031,
+      "step": 1160
+    },
+    {
+      "grad_norm": 0.6449679732322693,
+      "learning_rate": 9.878136768061154e-05,
+      "loss": 0.1985,
+      "step": 1170
+    },
+    {
+      "grad_norm": 0.619867742061615,
+      "learning_rate": 9.874481826960979e-05,
+      "loss": 0.1981,
+      "step": 1180
+    },
+    {
+      "grad_norm": 0.6148363351821899,
+      "learning_rate": 9.870773579308503e-05,
+      "loss": 0.1989,
+      "step": 1190
+    },
+    {
+      "grad_norm": 0.6031284928321838,
+      "learning_rate": 9.867012065656533e-05,
+      "loss": 0.2066,
+      "step": 1200
+    },
+    {
+      "grad_norm": 0.6985098719596863,
+      "learning_rate": 9.863197327140376e-05,
+      "loss": 0.2037,
+      "step": 1210
+    },
+    {
+      "grad_norm": 0.5774329304695129,
+      "learning_rate": 9.859329405477403e-05,
+      "loss": 0.1964,
+      "step": 1220
+    },
+    {
+      "grad_norm": 0.7787958979606628,
+      "learning_rate": 9.855408342966585e-05,
+      "loss": 0.1965,
+      "step": 1230
+    },
+    {
+      "grad_norm": 0.6748654246330261,
+      "learning_rate": 9.851434182488033e-05,
+      "loss": 0.1899,
+      "step": 1240
+    },
+    {
+      "grad_norm": 0.6843940019607544,
+      "learning_rate": 9.84740696750253e-05,
+      "loss": 0.1962,
+      "step": 1250
+    },
+    {
+      "grad_norm": 0.8092010617256165,
+      "learning_rate": 9.843326742051055e-05,
+      "loss": 0.188,
+      "step": 1260
+    },
+    {
+      "grad_norm": 0.6771940588951111,
+      "learning_rate": 9.839193550754297e-05,
+      "loss": 0.1922,
+      "step": 1270
+    },
+    {
+      "grad_norm": 0.5320672392845154,
+      "learning_rate": 9.835007438812177e-05,
+      "loss": 0.1855,
+      "step": 1280
+    },
+    {
+      "grad_norm": 0.710681140422821,
+      "learning_rate": 9.830768452003341e-05,
+      "loss": 0.1893,
+      "step": 1290
+    },
+    {
+      "grad_norm": 0.6156606674194336,
+      "learning_rate": 9.826476636684671e-05,
+      "loss": 0.1875,
+      "step": 1300
+    },
+    {
+      "grad_norm": 0.6500555872917175,
+      "learning_rate": 9.822132039790773e-05,
+      "loss": 0.1818,
+      "step": 1310
+    },
+    {
+      "grad_norm": 0.7215138077735901,
+      "learning_rate": 9.817734708833461e-05,
+      "loss": 0.1864,
+      "step": 1320
+    },
+    {
+      "grad_norm": 0.6235042214393616,
+      "learning_rate": 9.813284691901243e-05,
+      "loss": 0.1788,
+      "step": 1330
+    },
+    {
+      "grad_norm": 0.5805934071540833,
+      "learning_rate": 9.808782037658792e-05,
+      "loss": 0.1769,
+      "step": 1340
+    },
+    {
+      "grad_norm": 0.5326550602912903,
+      "learning_rate": 9.804226795346411e-05,
+      "loss": 0.1684,
+      "step": 1350
+    },
+    {
+      "grad_norm": 0.6381980180740356,
+      "learning_rate": 9.799619014779503e-05,
+      "loss": 0.1734,
+      "step": 1360
+    },
+    {
+      "grad_norm": 0.59572434425354,
+      "learning_rate": 9.794958746348013e-05,
+      "loss": 0.1665,
+      "step": 1370
+    },
+    {
+      "grad_norm": 0.530421257019043,
+      "learning_rate": 9.790246041015896e-05,
+      "loss": 0.1679,
+      "step": 1380
+    },
+    {
+      "grad_norm": 0.5774155259132385,
+      "learning_rate": 9.785480950320538e-05,
+      "loss": 0.1662,
+      "step": 1390
+    },
+    {
+      "grad_norm": 0.6030677556991577,
+      "learning_rate": 9.78066352637221e-05,
+      "loss": 0.1723,
+      "step": 1400
+    },
+    {
+      "grad_norm": 0.6545712947845459,
+      "learning_rate": 9.775793821853488e-05,
+      "loss": 0.1629,
+      "step": 1410
+    },
+    {
+      "grad_norm": 0.7584843635559082,
+      "learning_rate": 9.77087189001868e-05,
+      "loss": 0.1644,
+      "step": 1420
+    },
+    {
+      "grad_norm": 0.767001748085022,
+      "learning_rate": 9.765897784693243e-05,
+      "loss": 0.1603,
+      "step": 1430
+    },
+    {
+      "grad_norm": 0.5734662413597107,
+      "learning_rate": 9.760871560273197e-05,
+      "loss": 0.1563,
+      "step": 1440
+    },
+    {
+      "grad_norm": 0.7333984375,
+      "learning_rate": 9.755793271724526e-05,
+      "loss": 0.1609,
+      "step": 1450
+    },
+    {
+      "grad_norm": 0.5877911448478699,
+      "learning_rate": 9.750662974582584e-05,
+      "loss": 0.1564,
+      "step": 1460
+    },
+    {
+      "grad_norm": 0.6123472452163696,
+      "learning_rate": 9.745480724951473e-05,
+      "loss": 0.1568,
+      "step": 1470
+    },
+    {
+      "grad_norm": 0.6052020192146301,
+      "learning_rate": 9.740246579503447e-05,
+      "loss": 0.1598,
+      "step": 1480
+    },
+    {
+      "grad_norm": 0.6139975190162659,
+      "learning_rate": 9.734960595478284e-05,
+      "loss": 0.1509,
+      "step": 1490
+    },
+    {
+      "grad_norm": 0.6344926953315735,
+      "learning_rate": 9.729622830682657e-05,
+      "loss": 0.162,
+      "step": 1500
+    },
+    {
+      "grad_norm": 0.5783476233482361,
+      "learning_rate": 9.724233343489504e-05,
+      "loss": 0.1559,
+      "step": 1510
+    },
+    {
+      "grad_norm": 0.559226930141449,
+      "learning_rate": 9.718792192837396e-05,
+      "loss": 0.1549,
+      "step": 1520
+    },
+    {
+      "grad_norm": 0.645209014415741,
+      "learning_rate": 9.713299438229886e-05,
+      "loss": 0.1562,
+      "step": 1530
+    },
+    {
+      "grad_norm": 0.7524130344390869,
+      "learning_rate": 9.707755139734855e-05,
+      "loss": 0.1561,
+      "step": 1540
+    },
+    {
+      "grad_norm": 0.7448381185531616,
+      "learning_rate": 9.702159357983866e-05,
+      "loss": 0.1502,
+      "step": 1550
+    },
+    {
+      "grad_norm": 0.6631969809532166,
+      "learning_rate": 9.696512154171492e-05,
+      "loss": 0.1448,
+      "step": 1560
+    },
+    {
+      "grad_norm": 0.6108953356742859,
+      "learning_rate": 9.690813590054645e-05,
+      "loss": 0.1382,
+      "step": 1570
+    },
+    {
+      "grad_norm": 0.6539977192878723,
+      "learning_rate": 9.685063727951914e-05,
+      "loss": 0.138,
+      "step": 1580
+    },
+    {
+      "grad_norm": 0.6970216035842896,
+      "learning_rate": 9.679262630742865e-05,
+      "loss": 0.1382,
+      "step": 1590
+    },
+    {
+      "grad_norm": 0.824339747428894,
+      "learning_rate": 9.673410361867373e-05,
+      "loss": 0.1382,
+      "step": 1600
+    },
+    {
+      "grad_norm": 0.6064515709877014,
+      "learning_rate": 9.667506985324909e-05,
+      "loss": 0.1296,
+      "step": 1610
+    },
+    {
+      "grad_norm": 0.5242080688476562,
+      "learning_rate": 9.661552565673855e-05,
+      "loss": 0.1296,
+      "step": 1620
+    },
+    {
+      "grad_norm": 0.6478292346000671,
+      "learning_rate": 9.655547168030789e-05,
+      "loss": 0.1263,
+      "step": 1630
+    },
+    {
+      "grad_norm": 0.8052884936332703,
+      "learning_rate": 9.649490858069777e-05,
+      "loss": 0.128,
+      "step": 1640
+    },
+    {
+      "grad_norm": 0.6823225021362305,
+      "learning_rate": 9.643383702021658e-05,
+      "loss": 0.1373,
+      "step": 1650
+    },
+    {
+      "grad_norm": 0.49175184965133667,
+      "learning_rate": 9.637225766673307e-05,
+      "loss": 0.1202,
+      "step": 1660
+    },
+    {
+      "grad_norm": 0.5120248198509216,
+      "learning_rate": 9.631017119366922e-05,
+      "loss": 0.1254,
+      "step": 1670
+    },
+    {
+      "grad_norm": 0.688744068145752,
+      "learning_rate": 9.624757827999273e-05,
+      "loss": 0.133,
+      "step": 1680
+    },
+    {
+      "grad_norm": 0.5714597702026367,
+      "learning_rate": 9.618447961020971e-05,
+      "loss": 0.1312,
+      "step": 1690
+    },
+    {
+      "grad_norm": 0.6494192481040955,
+      "learning_rate": 9.612087587435707e-05,
+      "loss": 0.1251,
+      "step": 1700
+    },
+    {
+      "grad_norm": 0.5544602274894714,
+      "learning_rate": 9.605676776799508e-05,
+      "loss": 0.1295,
+      "step": 1710
+    },
+    {
+      "grad_norm": 0.6585219502449036,
+      "learning_rate": 9.599215599219973e-05,
+      "loss": 0.1292,
+      "step": 1720
+    },
+    {
+      "grad_norm": 0.6556674838066101,
+      "learning_rate": 9.592704125355505e-05,
+      "loss": 0.1293,
+      "step": 1730
+    },
+    {
+      "grad_norm": 0.601507306098938,
+      "learning_rate": 9.586142426414538e-05,
+      "loss": 0.1238,
+      "step": 1740
+    },
+    {
+      "grad_norm": 0.7637534141540527,
+      "learning_rate": 9.57953057415476e-05,
+      "loss": 0.1315,
+      "step": 1750
+    },
+    {
+      "grad_norm": 0.707006573677063,
+      "learning_rate": 9.572868640882328e-05,
+      "loss": 0.1288,
+      "step": 1760
+    },
+    {
+      "grad_norm": 0.6856809258460999,
+      "learning_rate": 9.56615669945108e-05,
+      "loss": 0.1266,
+      "step": 1770
+    },
+    {
+      "grad_norm": 0.5079591274261475,
+      "learning_rate": 9.55939482326173e-05,
+      "loss": 0.1218,
+      "step": 1780
+    },
+    {
+      "grad_norm": 0.7197537422180176,
+      "learning_rate": 9.552583086261069e-05,
+      "loss": 0.1188,
+      "step": 1790
+    },
+    {
+      "grad_norm": 0.7753213047981262,
+      "learning_rate": 9.545721562941168e-05,
+      "loss": 0.113,
+      "step": 1800
+    },
+    {
+      "grad_norm": 0.551957905292511,
+      "learning_rate": 9.538810328338543e-05,
+      "loss": 0.1097,
+      "step": 1810
+    },
+    {
+      "grad_norm": 0.5827521681785583,
+      "learning_rate": 9.531849458033349e-05,
+      "loss": 0.106,
+      "step": 1820
+    },
+    {
+      "grad_norm": 0.6864978075027466,
+      "learning_rate": 9.524839028148547e-05,
+      "loss": 0.1029,
+      "step": 1830
+    },
+    {
+      "grad_norm": 0.6032189130783081,
+      "learning_rate": 9.517779115349077e-05,
+      "loss": 0.1025,
+      "step": 1840
+    },
+    {
+      "grad_norm": 0.6585895419120789,
+      "learning_rate": 9.510669796841014e-05,
+      "loss": 0.1048,
+      "step": 1850
+    },
+    {
+      "grad_norm": 0.7293878793716431,
+      "learning_rate": 9.503511150370727e-05,
+      "loss": 0.1091,
+      "step": 1860
+    },
+    {
+      "grad_norm": 0.7068802118301392,
+      "learning_rate": 9.496303254224024e-05,
+      "loss": 0.1095,
+      "step": 1870
+    },
+    {
+      "grad_norm": 0.5836649537086487,
+      "learning_rate": 9.489046187225306e-05,
+      "loss": 0.0991,
+      "step": 1880
+    },
+    {
+      "grad_norm": 0.516670823097229,
+      "learning_rate": 9.481740028736692e-05,
+      "loss": 0.1004,
+      "step": 1890
+    },
+    {
+      "grad_norm": 0.5521473288536072,
+      "learning_rate": 9.474384858657164e-05,
+      "loss": 0.1167,
+      "step": 1900
+    },
+    {
+      "grad_norm": 0.7428975701332092,
+      "learning_rate": 9.466980757421679e-05,
+      "loss": 0.1091,
+      "step": 1910
+    },
+    {
+      "grad_norm": 0.585248589515686,
+      "learning_rate": 9.459527806000305e-05,
+      "loss": 0.1098,
+      "step": 1920
+    },
+    {
+      "grad_norm": 0.7214521169662476,
+      "learning_rate": 9.452026085897325e-05,
+      "loss": 0.109,
+      "step": 1930
+    },
+    {
+      "grad_norm": 0.6236957907676697,
+      "learning_rate": 9.444475679150348e-05,
+      "loss": 0.1056,
+      "step": 1940
+    },
+    {
+      "grad_norm": 0.5456467866897583,
+      "learning_rate": 9.436876668329411e-05,
+      "loss": 0.1054,
+      "step": 1950
+    },
+    {
+      "grad_norm": 0.5371605753898621,
+      "learning_rate": 9.429229136536079e-05,
+      "loss": 0.1019,
+      "step": 1960
+    },
+    {
+      "grad_norm": 0.49974414706230164,
+      "learning_rate": 9.421533167402534e-05,
+      "loss": 0.1064,
+      "step": 1970
+    },
+    {
+      "grad_norm": 0.6514241695404053,
+      "learning_rate": 9.413788845090666e-05,
+      "loss": 0.1125,
+      "step": 1980
+    },
+    {
+      "grad_norm": 0.7077551484107971,
+      "learning_rate": 9.405996254291136e-05,
+      "loss": 0.1046,
+      "step": 1990
+    },
+    {
+      "grad_norm": 0.5884621739387512,
+      "learning_rate": 9.398155480222474e-05,
+      "loss": 0.1038,
+      "step": 2000
+    },
+    {
+      "grad_norm": 0.5488919615745544,
+      "learning_rate": 9.390266608630128e-05,
+      "loss": 0.0994,
+      "step": 2010
+    },
+    {
+      "grad_norm": 0.6028702855110168,
+      "learning_rate": 9.38232972578553e-05,
+      "loss": 0.1024,
+      "step": 2020
+    },
+    {
+      "grad_norm": 0.6464540362358093,
+      "learning_rate": 9.374344918485164e-05,
+      "loss": 0.103,
+      "step": 2030
+    },
+    {
+      "grad_norm": 0.5217083096504211,
+      "learning_rate": 9.366312274049602e-05,
+      "loss": 0.1109,
+      "step": 2040
+    },
+    {
+      "grad_norm": 0.6009257435798645,
+      "learning_rate": 9.358231880322554e-05,
+      "loss": 0.0978,
+      "step": 2050
+    },
+    {
+      "grad_norm": 0.572211503982544,
+      "learning_rate": 9.350103825669916e-05,
+      "loss": 0.0974,
+      "step": 2060
+    },
+    {
+      "grad_norm": 0.5632114410400391,
+      "learning_rate": 9.341928198978787e-05,
+      "loss": 0.0995,
+      "step": 2070
+    },
+    {
+      "grad_norm": 0.5880551338195801,
+      "learning_rate": 9.333705089656512e-05,
+      "loss": 0.0929,
+      "step": 2080
+    },
+    {
+      "grad_norm": 0.6269182562828064,
+      "learning_rate": 9.325434587629698e-05,
+      "loss": 0.0991,
+      "step": 2090
+    },
+    {
+      "grad_norm": 0.5491243600845337,
+      "learning_rate": 9.31711678334323e-05,
+      "loss": 0.0926,
+      "step": 2100
+    },
+    {
+      "grad_norm": 0.4559735059738159,
+      "learning_rate": 9.308751767759282e-05,
+      "loss": 0.0939,
+      "step": 2110
+    },
+    {
+      "grad_norm": 0.7348751425743103,
+      "learning_rate": 9.300339632356325e-05,
+      "loss": 0.0909,
+      "step": 2120
+    },
+    {
+      "grad_norm": 0.6080466508865356,
+      "learning_rate": 9.291880469128124e-05,
+      "loss": 0.0918,
+      "step": 2130
+    },
+    {
+      "grad_norm": 0.743401050567627,
+      "learning_rate": 9.283374370582732e-05,
+      "loss": 0.0924,
+      "step": 2140
+    },
+    {
+      "grad_norm": 0.6211663484573364,
+      "learning_rate": 9.274821429741482e-05,
+      "loss": 0.1029,
+      "step": 2150
+    },
+    {
+      "grad_norm": 0.7071795463562012,
+      "learning_rate": 9.266221740137961e-05,
+      "loss": 0.0923,
+      "step": 2160
+    },
+    {
+      "grad_norm": 0.7179328799247742,
+      "learning_rate": 9.257575395817001e-05,
+      "loss": 0.0804,
+      "step": 2170
+    },
+    {
+      "grad_norm": 0.8044621348381042,
+      "learning_rate": 9.248882491333637e-05,
+      "loss": 0.0843,
+      "step": 2180
+    },
+    {
+      "grad_norm": 0.6661281585693359,
+      "learning_rate": 9.240143121752076e-05,
+      "loss": 0.0855,
+      "step": 2190
+    },
+    {
+      "grad_norm": 0.5799790024757385,
+      "learning_rate": 9.23135738264467e-05,
+      "loss": 0.0795,
+      "step": 2200
+    },
+    {
+      "grad_norm": 0.7222135066986084,
+      "learning_rate": 9.222525370090849e-05,
+      "loss": 0.0751,
+      "step": 2210
+    },
+    {
+      "grad_norm": 0.6246083974838257,
+      "learning_rate": 9.213647180676088e-05,
+      "loss": 0.0769,
+      "step": 2220
+    },
+    {
+      "grad_norm": 0.6597115397453308,
+      "learning_rate": 9.204722911490846e-05,
+      "loss": 0.0824,
+      "step": 2230
+    },
+    {
+      "grad_norm": 0.5377702713012695,
+      "learning_rate": 9.1957526601295e-05,
+      "loss": 0.0806,
+      "step": 2240
+    },
+    {
+      "grad_norm": 0.5512954592704773,
+      "learning_rate": 9.186736524689281e-05,
+      "loss": 0.0819,
+      "step": 2250
+    },
+    {
+      "grad_norm": 0.6828509569168091,
+      "learning_rate": 9.177674603769204e-05,
+      "loss": 0.0794,
+      "step": 2260
+    },
+    {
+      "grad_norm": 0.6213656663894653,
+      "learning_rate": 9.168566996468983e-05,
+      "loss": 0.0762,
+      "step": 2270
+    },
+    {
+      "grad_norm": 0.6774435639381409,
+      "learning_rate": 9.159413802387951e-05,
+      "loss": 0.0795,
+      "step": 2280
+    },
+    {
+      "grad_norm": 0.6063661575317383,
+      "learning_rate": 9.150215121623974e-05,
+      "loss": 0.0786,
+      "step": 2290
+    },
+    {
+      "grad_norm": 0.4777480363845825,
+      "learning_rate": 9.140971054772349e-05,
+      "loss": 0.077,
+      "step": 2300
+    },
+    {
+      "grad_norm": 0.6887581944465637,
+      "learning_rate": 9.131681702924713e-05,
+      "loss": 0.0842,
+      "step": 2310
+    },
+    {
+      "grad_norm": 0.5813681483268738,
+      "learning_rate": 9.122347167667926e-05,
+      "loss": 0.077,
+      "step": 2320
+    },
+    {
+      "grad_norm": 0.5324981212615967,
+      "learning_rate": 9.112967551082973e-05,
+      "loss": 0.0826,
+      "step": 2330
+    },
+    {
+      "grad_norm": 0.6038345694541931,
+      "learning_rate": 9.103542955743835e-05,
+      "loss": 0.0751,
+      "step": 2340
+    },
+    {
+      "grad_norm": 0.6981058716773987,
+      "learning_rate": 9.094073484716381e-05,
+      "loss": 0.0794,
+      "step": 2350
+    },
+    {
+      "grad_norm": 0.7169125080108643,
+      "learning_rate": 9.084559241557226e-05,
+      "loss": 0.0717,
+      "step": 2360
+    },
+    {
+      "grad_norm": 0.7259419560432434,
+      "learning_rate": 9.075000330312608e-05,
+      "loss": 0.0688,
+      "step": 2370
+    },
+    {
+      "grad_norm": 0.718980073928833,
+      "learning_rate": 9.065396855517253e-05,
+      "loss": 0.0631,
+      "step": 2380
+    },
+    {
+      "grad_norm": 0.7885380387306213,
+      "learning_rate": 9.055748922193219e-05,
+      "loss": 0.0704,
+      "step": 2390
+    },
+    {
+      "grad_norm": 0.6676114797592163,
+      "learning_rate": 9.046056635848761e-05,
+      "loss": 0.0693,
+      "step": 2400
+    },
+    {
+      "grad_norm": 0.6262313723564148,
+      "learning_rate": 9.036320102477169e-05,
+      "loss": 0.0579,
+      "step": 2410
+    },
+    {
+      "grad_norm": 0.6463766098022461,
+      "learning_rate": 9.02653942855561e-05,
+      "loss": 0.0636,
+      "step": 2420
+    },
+    {
+      "grad_norm": 0.6266476511955261,
+      "learning_rate": 9.016714721043971e-05,
+      "loss": 0.058,
+      "step": 2430
+    },
+    {
+      "grad_norm": 0.6676295399665833,
+      "learning_rate": 9.006846087383675e-05,
+      "loss": 0.0612,
+      "step": 2440
+    },
+    {
+      "grad_norm": 0.6078400015830994,
+      "learning_rate": 8.996933635496523e-05,
+      "loss": 0.0583,
+      "step": 2450
+    },
+    {
+      "grad_norm": 0.6385151743888855,
+      "learning_rate": 8.986977473783498e-05,
+      "loss": 0.0635,
+      "step": 2460
+    },
+    {
+      "grad_norm": 0.5878050923347473,
+      "learning_rate": 8.97697771112359e-05,
+      "loss": 0.0576,
+      "step": 2470
+    },
+    {
+      "grad_norm": 0.6868295073509216,
+      "learning_rate": 8.966934456872602e-05,
+      "loss": 0.0592,
+      "step": 2480
+    },
+    {
+      "grad_norm": 0.5964924693107605,
+      "learning_rate": 8.95684782086195e-05,
+      "loss": 0.0492,
+      "step": 2490
+    },
+    {
+      "grad_norm": 0.5749147534370422,
+      "learning_rate": 8.946717913397476e-05,
+      "loss": 0.0609,
+      "step": 2500
+    },
+    {
+      "grad_norm": 0.6247969269752502,
+      "learning_rate": 8.93654484525822e-05,
+      "loss": 0.058,
+      "step": 2510
+    },
+    {
+      "grad_norm": 0.5573066473007202,
+      "learning_rate": 8.926328727695226e-05,
+      "loss": 0.0572,
+      "step": 2520
+    },
+    {
+      "grad_norm": 0.5724696516990662,
+      "learning_rate": 8.916069672430319e-05,
+      "loss": 0.052,
+      "step": 2530
+    },
+    {
+      "grad_norm": 0.5338303446769714,
+      "learning_rate": 8.905767791654884e-05,
+      "loss": 0.0582,
+      "step": 2540
+    },
+    {
+      "grad_norm": 0.7455699443817139,
+      "learning_rate": 8.895423198028638e-05,
+      "loss": 0.0662,
+      "step": 2550
+    },
+    {
+      "grad_norm": 0.5584907531738281,
+      "learning_rate": 8.885036004678402e-05,
+      "loss": 0.0563,
+      "step": 2560
+    },
+    {
+      "grad_norm": 0.599135160446167,
+      "learning_rate": 8.874606325196857e-05,
+      "loss": 0.0561,
+      "step": 2570
+    },
+    {
+      "grad_norm": 0.5163583755493164,
+      "learning_rate": 8.864134273641304e-05,
+      "loss": 0.0544,
+      "step": 2580
+    },
+    {
+      "grad_norm": 0.6540436148643494,
+      "learning_rate": 8.853619964532427e-05,
+      "loss": 0.0621,
+      "step": 2590
+    },
+    {
+      "grad_norm": 0.6071038246154785,
+      "learning_rate": 8.843063512853019e-05,
+      "loss": 0.0528,
+      "step": 2600
+    },
+    {
+      "grad_norm": 0.48813730478286743,
+      "learning_rate": 8.832465034046749e-05,
+      "loss": 0.0528,
+      "step": 2610
+    },
+    {
+      "grad_norm": 0.5264520645141602,
+      "learning_rate": 8.821824644016882e-05,
+      "loss": 0.0521,
+      "step": 2620
+    },
+    {
+      "grad_norm": 0.5260543823242188,
+      "learning_rate": 8.811142459125019e-05,
+      "loss": 0.0525,
+      "step": 2630
+    },
+    {
+      "grad_norm": 0.563161313533783,
+      "learning_rate": 8.800418596189822e-05,
+      "loss": 0.0478,
+      "step": 2640
+    },
+    {
+      "grad_norm": 0.5909938812255859,
+      "learning_rate": 8.789653172485737e-05,
+      "loss": 0.0556,
+      "step": 2650
+    },
+    {
+      "grad_norm": 0.7121275663375854,
+      "learning_rate": 8.778846305741715e-05,
+      "loss": 0.0658,
+      "step": 2660
+    },
+    {
+      "grad_norm": 0.46030551195144653,
+      "learning_rate": 8.767998114139918e-05,
+      "loss": 0.0584,
+      "step": 2670
+    },
+    {
+      "grad_norm": 0.6642009615898132,
+      "learning_rate": 8.757108716314429e-05,
+      "loss": 0.0623,
+      "step": 2680
+    },
+    {
+      "grad_norm": 0.5153357982635498,
+      "learning_rate": 8.746178231349962e-05,
+      "loss": 0.0543,
+      "step": 2690
+    },
+    {
+      "grad_norm": 0.5011047720909119,
+      "learning_rate": 8.735206778780549e-05,
+      "loss": 0.0568,
+      "step": 2700
+    },
+    {
+      "grad_norm": 0.4046159088611603,
+      "learning_rate": 8.724194478588234e-05,
+      "loss": 0.056,
+      "step": 2710
+    },
+    {
+      "grad_norm": 0.5637795925140381,
+      "learning_rate": 8.713141451201772e-05,
+      "loss": 0.0515,
+      "step": 2720
+    },
+    {
+      "grad_norm": 0.584834098815918,
+      "learning_rate": 8.702047817495295e-05,
+      "loss": 0.0522,
+      "step": 2730
+    },
+    {
+      "grad_norm": 0.5140633583068848,
+      "learning_rate": 8.69091369878701e-05,
+      "loss": 0.0541,
+      "step": 2740
+    },
+    {
+      "grad_norm": 0.5771273374557495,
+      "learning_rate": 8.679739216837849e-05,
+      "loss": 0.051,
+      "step": 2750
+    },
+    {
+      "grad_norm": 0.5800580978393555,
+      "learning_rate": 8.66852449385016e-05,
+      "loss": 0.051,
+      "step": 2760
+    },
+    {
+      "grad_norm": 0.5376407504081726,
+      "learning_rate": 8.657269652466356e-05,
+      "loss": 0.0482,
+      "step": 2770
+    },
+    {
+      "grad_norm": 0.5004979372024536,
+      "learning_rate": 8.645974815767577e-05,
+      "loss": 0.05,
+      "step": 2780
+    },
+    {
+      "grad_norm": 0.7781835198402405,
+      "learning_rate": 8.634640107272351e-05,
+      "loss": 0.0512,
+      "step": 2790
+    },
+    {
+      "grad_norm": 0.7080514430999756,
+      "learning_rate": 8.623265650935234e-05,
+      "loss": 0.0501,
+      "step": 2800
+    },
+    {
+      "grad_norm": 0.4747622013092041,
+      "learning_rate": 8.611851571145456e-05,
+      "loss": 0.0554,
+      "step": 2810
+    },
+    {
+      "grad_norm": 0.4781681299209595,
+      "learning_rate": 8.600397992725566e-05,
+      "loss": 0.0512,
+      "step": 2820
+    },
+    {
+      "grad_norm": 0.4888535737991333,
+      "learning_rate": 8.588905040930061e-05,
+      "loss": 0.0502,
+      "step": 2830
+    },
+    {
+      "grad_norm": 0.4943699538707733,
+      "learning_rate": 8.577372841444022e-05,
+      "loss": 0.0501,
+      "step": 2840
+    },
+    {
+      "grad_norm": 0.5830317139625549,
+      "learning_rate": 8.565801520381736e-05,
+      "loss": 0.0489,
+      "step": 2850
+    },
+    {
+      "grad_norm": 0.5344899296760559,
+      "learning_rate": 8.554191204285313e-05,
+      "loss": 0.0525,
+      "step": 2860
+    },
+    {
+      "grad_norm": 0.46698901057243347,
+      "learning_rate": 8.542542020123315e-05,
+      "loss": 0.0519,
+      "step": 2870
+    },
+    {
+      "grad_norm": 0.565811038017273,
+      "learning_rate": 8.530854095289347e-05,
+      "loss": 0.0508,
+      "step": 2880
+    },
+    {
+      "grad_norm": 0.6035529971122742,
+      "learning_rate": 8.519127557600688e-05,
+      "loss": 0.0468,
+      "step": 2890
+    },
+    {
+      "grad_norm": 0.7289096117019653,
+      "learning_rate": 8.507362535296871e-05,
+      "loss": 0.0484,
+      "step": 2900
+    },
+    {
+      "grad_norm": 0.5606696605682373,
+      "learning_rate": 8.495559157038299e-05,
+      "loss": 0.0652,
+      "step": 2910
+    },
+    {
+      "grad_norm": 0.6256048679351807,
+      "learning_rate": 8.483717551904823e-05,
+      "loss": 0.0539,
+      "step": 2920
+    },
+    {
+      "grad_norm": 0.39223653078079224,
+      "learning_rate": 8.47183784939434e-05,
+      "loss": 0.0477,
+      "step": 2930
+    },
+    {
+      "grad_norm": 0.46564576029777527,
+      "learning_rate": 8.459920179421374e-05,
+      "loss": 0.0482,
+      "step": 2940
+    },
+    {
+      "grad_norm": 0.5675666332244873,
+      "learning_rate": 8.447964672315656e-05,
+      "loss": 0.0458,
+      "step": 2950
+    },
+    {
+      "grad_norm": 0.5843992829322815,
+      "learning_rate": 8.435971458820692e-05,
+      "loss": 0.0541,
+      "step": 2960
+    },
+    {
+      "grad_norm": 0.5688618421554565,
+      "learning_rate": 8.423940670092345e-05,
+      "loss": 0.0525,
+      "step": 2970
+    },
+    {
+      "grad_norm": 0.5131046175956726,
+      "learning_rate": 8.411872437697394e-05,
+      "loss": 0.0488,
+      "step": 2980
+    },
+    {
+      "grad_norm": 2.3579087257385254,
+      "learning_rate": 8.399766893612096e-05,
+      "loss": 0.0472,
+      "step": 2990
+    },
+    {
+      "grad_norm": 1.2814431190490723,
+      "learning_rate": 8.38762417022074e-05,
+      "loss": 0.0547,
+      "step": 3000
+    },
+    {
+      "grad_norm": 0.6536903381347656,
+      "learning_rate": 8.375444400314204e-05,
+      "loss": 0.0534,
+      "step": 3010
+    },
+    {
+      "grad_norm": 0.5091704726219177,
+      "learning_rate": 8.3632277170885e-05,
+      "loss": 0.0593,
+      "step": 3020
+    },
+    {
+      "grad_norm": 0.5296808481216431,
+      "learning_rate": 8.350974254143318e-05,
+      "loss": 0.0458,
+      "step": 3030
+    },
+    {
+      "grad_norm": 0.5711715221405029,
+      "learning_rate": 8.338684145480566e-05,
+      "loss": 0.0478,
+      "step": 3040
+    },
+    {
+      "grad_norm": 0.6120307445526123,
+      "learning_rate": 8.326357525502904e-05,
+      "loss": 0.0552,
+      "step": 3050
+    },
+    {
+      "grad_norm": 0.4970693290233612,
+      "learning_rate": 8.313994529012273e-05,
+      "loss": 0.05,
+      "step": 3060
+    },
+    {
+      "grad_norm": 0.45827317237854004,
+      "learning_rate": 8.301595291208422e-05,
+      "loss": 0.0551,
+      "step": 3070
+    },
+    {
+      "grad_norm": 0.5493773818016052,
+      "learning_rate": 8.289159947687427e-05,
+      "loss": 0.0504,
+      "step": 3080
+    },
+    {
+      "grad_norm": 0.5398691892623901,
+      "learning_rate": 8.276688634440216e-05,
+      "loss": 0.0507,
+      "step": 3090
+    },
+    {
+      "grad_norm": 0.4500541090965271,
+      "learning_rate": 8.26418148785107e-05,
+      "loss": 0.0486,
+      "step": 3100
+    },
+    {
+      "grad_norm": 0.6306096911430359,
+      "learning_rate": 8.251638644696141e-05,
+      "loss": 0.0517,
+      "step": 3110
+    },
+    {
+      "grad_norm": 0.44671720266342163,
+      "learning_rate": 8.23906024214195e-05,
+      "loss": 0.0516,
+      "step": 3120
+    },
+    {
+      "grad_norm": 0.47543811798095703,
+      "learning_rate": 8.226446417743897e-05,
+      "loss": 0.0496,
+      "step": 3130
+    },
+    {
+      "grad_norm": 0.580166220664978,
+      "learning_rate": 8.213797309444742e-05,
+      "loss": 0.0495,
+      "step": 3140
+    },
+    {
+      "grad_norm": 0.6284545063972473,
+      "learning_rate": 8.201113055573105e-05,
+      "loss": 0.0493,
+      "step": 3150
+    },
+    {
+      "grad_norm": 0.5596886873245239,
+      "learning_rate": 8.188393794841958e-05,
+      "loss": 0.0465,
+      "step": 3160
+    },
+    {
+      "grad_norm": 0.5735827684402466,
+      "learning_rate": 8.175639666347094e-05,
+      "loss": 0.0536,
+      "step": 3170
+    },
+    {
+      "grad_norm": 0.5282607674598694,
+      "learning_rate": 8.162850809565623e-05,
+      "loss": 0.0471,
+      "step": 3180
+    },
+    {
+      "grad_norm": 0.572863757610321,
+      "learning_rate": 8.150027364354431e-05,
+      "loss": 0.0483,
+      "step": 3190
+    },
+    {
+      "grad_norm": 0.6053555011749268,
+      "learning_rate": 8.137169470948662e-05,
+      "loss": 0.0537,
+      "step": 3200
+    },
+    {
+      "grad_norm": 0.6028903126716614,
+      "learning_rate": 8.124277269960179e-05,
+      "loss": 0.0518,
+      "step": 3210
+    },
+    {
+      "grad_norm": 0.4950304925441742,
+      "learning_rate": 8.111350902376023e-05,
+      "loss": 0.0489,
+      "step": 3220
+    },
+    {
+      "grad_norm": 0.4918529987335205,
+      "learning_rate": 8.098390509556883e-05,
+      "loss": 0.0478,
+      "step": 3230
+    },
+    {
+      "grad_norm": 0.5505935549736023,
+      "learning_rate": 8.085396233235536e-05,
+      "loss": 0.044,
+      "step": 3240
+    },
+    {
+      "grad_norm": 0.517212450504303,
+      "learning_rate": 8.072368215515306e-05,
+      "loss": 0.0442,
+      "step": 3250
+    },
+    {
+      "grad_norm": 0.6040185689926147,
+      "learning_rate": 8.059306598868506e-05,
+      "loss": 0.0554,
+      "step": 3260
+    },
+    {
+      "grad_norm": 0.4814482033252716,
+      "learning_rate": 8.046211526134888e-05,
+      "loss": 0.0491,
+      "step": 3270
+    },
+    {
+      "grad_norm": 0.38651132583618164,
+      "learning_rate": 8.033083140520065e-05,
+      "loss": 0.0439,
+      "step": 3280
+    },
+    {
+      "grad_norm": 0.4198136329650879,
+      "learning_rate": 8.019921585593962e-05,
+      "loss": 0.0494,
+      "step": 3290
+    },
+    {
+      "grad_norm": 0.5512520670890808,
+      "learning_rate": 8.006727005289232e-05,
+      "loss": 0.0451,
+      "step": 3300
+    },
+    {
+      "grad_norm": 0.5029220581054688,
+      "learning_rate": 7.993499543899692e-05,
+      "loss": 0.0457,
+      "step": 3310
+    },
+    {
+      "grad_norm": 0.5571017265319824,
+      "learning_rate": 7.980239346078742e-05,
+      "loss": 0.0414,
+      "step": 3320
+    },
+    {
+      "grad_norm": 0.5938437581062317,
+      "learning_rate": 7.966946556837778e-05,
+      "loss": 0.0469,
+      "step": 3330
+    },
+    {
+      "grad_norm": 0.8012939095497131,
+      "learning_rate": 7.953621321544616e-05,
+      "loss": 0.0489,
+      "step": 3340
+    },
+    {
+      "grad_norm": 0.6161916851997375,
+      "learning_rate": 7.940263785921896e-05,
+      "loss": 0.0537,
+      "step": 3350
+    },
+    {
+      "grad_norm": 0.6651037335395813,
+      "learning_rate": 7.926874096045482e-05,
+      "loss": 0.0431,
+      "step": 3360
+    },
+    {
+      "grad_norm": 0.5800148844718933,
+      "learning_rate": 7.913452398342881e-05,
+      "loss": 0.0573,
+      "step": 3370
+    },
+    {
+      "grad_norm": 0.5264511704444885,
+      "learning_rate": 7.89999883959163e-05,
+      "loss": 0.0528,
+      "step": 3380
+    },
+    {
+      "grad_norm": 0.6768236756324768,
+      "learning_rate": 7.886513566917687e-05,
+      "loss": 0.0536,
+      "step": 3390
+    },
+    {
+      "grad_norm": 0.6964791417121887,
+      "learning_rate": 7.872996727793838e-05,
+      "loss": 0.0567,
+      "step": 3400
+    },
+    {
+      "grad_norm": 0.5873032212257385,
+      "learning_rate": 7.859448470038069e-05,
+      "loss": 0.0512,
+      "step": 3410
+    },
+    {
+      "grad_norm": 0.6290404200553894,
+      "learning_rate": 7.845868941811956e-05,
+      "loss": 0.0477,
+      "step": 3420
+    },
+    {
+      "grad_norm": 0.5438164472579956,
+      "learning_rate": 7.832258291619043e-05,
+      "loss": 0.0484,
+      "step": 3430
+    },
+    {
+      "grad_norm": 0.6357219815254211,
+      "learning_rate": 7.81861666830322e-05,
+      "loss": 0.0504,
+      "step": 3440
+    },
+    {
+      "grad_norm": 0.5126264691352844,
+      "learning_rate": 7.804944221047097e-05,
+      "loss": 0.0463,
+      "step": 3450
+    },
+    {
+      "grad_norm": 0.44972777366638184,
+      "learning_rate": 7.791241099370364e-05,
+      "loss": 0.047,
+      "step": 3460
+    },
+    {
+      "grad_norm": 0.4316508173942566,
+      "learning_rate": 7.777507453128163e-05,
+      "loss": 0.0442,
+      "step": 3470
+    },
+    {
+      "grad_norm": 0.3861590027809143,
+      "learning_rate": 7.763743432509451e-05,
+      "loss": 0.0479,
+      "step": 3480
+    },
+    {
+      "grad_norm": 0.47488558292388916,
+      "learning_rate": 7.749949188035353e-05,
+      "loss": 0.0429,
+      "step": 3490
+    },
+    {
+      "grad_norm": 0.5337404608726501,
+      "learning_rate": 7.736124870557516e-05,
+      "loss": 0.0535,
+      "step": 3500
+    },
+    {
+      "grad_norm": 0.5130584239959717,
+      "learning_rate": 7.722270631256459e-05,
+      "loss": 0.0475,
+      "step": 3510
+    },
+    {
+      "grad_norm": 0.5091937184333801,
+      "learning_rate": 7.708386621639925e-05,
+      "loss": 0.0467,
+      "step": 3520
+    },
+    {
+      "grad_norm": 0.43654540181159973,
+      "learning_rate": 7.694472993541219e-05,
+      "loss": 0.0432,
+      "step": 3530
+    },
+    {
+      "grad_norm": 0.5237100720405579,
+      "learning_rate": 7.680529899117547e-05,
+      "loss": 0.0506,
+      "step": 3540
+    },
+    {
+      "grad_norm": 0.5649544596672058,
+      "learning_rate": 7.666557490848358e-05,
+      "loss": 0.0431,
+      "step": 3550
+    },
+    {
+      "grad_norm": 0.5605695247650146,
+      "learning_rate": 7.65255592153367e-05,
+      "loss": 0.0437,
+      "step": 3560
+    },
+    {
+      "grad_norm": 0.6432333588600159,
+      "learning_rate": 7.638525344292402e-05,
+      "loss": 0.0466,
+      "step": 3570
+    },
+    {
+      "grad_norm": 0.4990709125995636,
+      "learning_rate": 7.624465912560697e-05,
+      "loss": 0.0468,
+      "step": 3580
+    },
+    {
+      "grad_norm": 0.5626299381256104,
+      "learning_rate": 7.610377780090249e-05,
+      "loss": 0.0468,
+      "step": 3590
+    },
+    {
+      "grad_norm": 0.4099712669849396,
+      "learning_rate": 7.596261100946618e-05,
+      "loss": 0.0486,
+      "step": 3600
+    },
+    {
+      "grad_norm": 0.381757527589798,
+      "learning_rate": 7.582116029507542e-05,
+      "loss": 0.0403,
+      "step": 3610
+    },
+    {
+      "grad_norm": 0.4739331901073456,
+      "learning_rate": 7.56794272046126e-05,
+      "loss": 0.0436,
+      "step": 3620
+    },
+    {
+      "grad_norm": 0.3989870250225067,
+      "learning_rate": 7.55374132880481e-05,
+      "loss": 0.0439,
+      "step": 3630
+    },
+    {
+      "grad_norm": 0.42159515619277954,
+      "learning_rate": 7.539512009842333e-05,
+      "loss": 0.0469,
+      "step": 3640
+    },
+    {
+      "grad_norm": 0.5128747820854187,
+      "learning_rate": 7.525254919183382e-05,
+      "loss": 0.0516,
+      "step": 3650
+    },
+    {
+      "grad_norm": 0.43955498933792114,
+      "learning_rate": 7.510970212741215e-05,
+      "loss": 0.0427,
+      "step": 3660
+    },
+    {
+      "grad_norm": 0.46188902854919434,
+      "learning_rate": 7.496658046731096e-05,
+      "loss": 0.0456,
+      "step": 3670
+    },
+    {
+      "grad_norm": 0.5381562113761902,
+      "learning_rate": 7.482318577668578e-05,
+      "loss": 0.0488,
+      "step": 3680
+    },
+    {
+      "grad_norm": 0.5339709520339966,
+      "learning_rate": 7.467951962367796e-05,
+      "loss": 0.0456,
+      "step": 3690
+    },
+    {
+      "grad_norm": 0.5261500477790833,
+      "learning_rate": 7.453558357939755e-05,
+      "loss": 0.043,
+      "step": 3700
+    },
+    {
+      "grad_norm": 0.4030408561229706,
+      "learning_rate": 7.439137921790606e-05,
+      "loss": 0.0475,
+      "step": 3710
+    },
+    {
+      "grad_norm": 0.5079669952392578,
+      "learning_rate": 7.42469081161993e-05,
+      "loss": 0.0429,
+      "step": 3720
+    },
+    {
+      "grad_norm": 0.45403552055358887,
+      "learning_rate": 7.410217185419006e-05,
+      "loss": 0.0384,
+      "step": 3730
+    },
+    {
+      "grad_norm": 0.6405506134033203,
+      "learning_rate": 7.395717201469095e-05,
+      "loss": 0.0472,
+      "step": 3740
+    },
+    {
+      "grad_norm": 0.5707180500030518,
+      "learning_rate": 7.381191018339696e-05,
+      "loss": 0.0445,
+      "step": 3750
+    },
+    {
+      "grad_norm": 0.506778359413147,
+      "learning_rate": 7.36663879488682e-05,
+      "loss": 0.0427,
+      "step": 3760
+    },
+    {
+      "grad_norm": 0.4757100045681,
+      "learning_rate": 7.352060690251254e-05,
+      "loss": 0.0493,
+      "step": 3770
+    },
+    {
+      "grad_norm": 0.6890202760696411,
+      "learning_rate": 7.337456863856811e-05,
+      "loss": 0.0429,
+      "step": 3780
+    },
+    {
+      "grad_norm": 0.5848238468170166,
+      "learning_rate": 7.3228274754086e-05,
+      "loss": 0.0436,
+      "step": 3790
+    },
+    {
+      "grad_norm": 0.49470770359039307,
+      "learning_rate": 7.308172684891267e-05,
+      "loss": 0.0456,
+      "step": 3800
+    },
+    {
+      "grad_norm": 0.47321856021881104,
+      "learning_rate": 7.293492652567255e-05,
+      "loss": 0.0376,
+      "step": 3810
+    },
+    {
+      "grad_norm": 0.47240787744522095,
+      "learning_rate": 7.278787538975043e-05,
+      "loss": 0.0466,
+      "step": 3820
+    },
+    {
+      "grad_norm": 0.4964098632335663,
+      "learning_rate": 7.2640575049274e-05,
+      "loss": 0.0472,
+      "step": 3830
+    },
+    {
+      "grad_norm": 0.4397500157356262,
+      "learning_rate": 7.249302711509616e-05,
+      "loss": 0.0419,
+      "step": 3840
+    },
+    {
+      "grad_norm": 0.5331586003303528,
+      "learning_rate": 7.23452332007775e-05,
+      "loss": 0.0446,
+      "step": 3850
+    },
+    {
+      "grad_norm": 0.5101559162139893,
+      "learning_rate": 7.219719492256858e-05,
+      "loss": 0.0468,
+      "step": 3860
+    },
+    {
+      "grad_norm": 0.6204541921615601,
+      "learning_rate": 7.20489138993923e-05,
+      "loss": 0.0469,
+      "step": 3870
+    },
+    {
+      "grad_norm": 0.45076897740364075,
+      "learning_rate": 7.190039175282614e-05,
+      "loss": 0.0457,
+      "step": 3880
+    },
+    {
+      "grad_norm": 0.48851674795150757,
+      "learning_rate": 7.175163010708455e-05,
+      "loss": 0.0438,
+      "step": 3890
+    },
+    {
+      "grad_norm": 0.4611389636993408,
+      "learning_rate": 7.1602630589001e-05,
+      "loss": 0.0464,
+      "step": 3900
+    },
+    {
+      "grad_norm": 0.4361285865306854,
+      "learning_rate": 7.14533948280104e-05,
+      "loss": 0.0415,
+      "step": 3910
+    },
+    {
+      "grad_norm": 0.4512380063533783,
+      "learning_rate": 7.130392445613109e-05,
+      "loss": 0.0411,
+      "step": 3920
+    },
+    {
+      "grad_norm": 0.5638014674186707,
+      "learning_rate": 7.115422110794711e-05,
+      "loss": 0.0408,
+      "step": 3930
+    },
+    {
+      "grad_norm": 0.4771179258823395,
+      "learning_rate": 7.100428642059033e-05,
+      "loss": 0.0436,
+      "step": 3940
+    },
+    {
+      "grad_norm": 0.4358355700969696,
+      "learning_rate": 7.08541220337224e-05,
+      "loss": 0.049,
+      "step": 3950
+    },
+    {
+      "grad_norm": 0.5044130086898804,
+      "learning_rate": 7.070372958951706e-05,
+      "loss": 0.0408,
+      "step": 3960
+    },
+    {
+      "grad_norm": 0.5154502391815186,
+      "learning_rate": 7.055311073264194e-05,
+      "loss": 0.0435,
+      "step": 3970
+    },
+    {
+      "grad_norm": 0.6458137631416321,
+      "learning_rate": 7.040226711024077e-05,
+      "loss": 0.042,
+      "step": 3980
+    },
+    {
+      "grad_norm": 0.5193631052970886,
+      "learning_rate": 7.02512003719152e-05,
+      "loss": 0.0379,
+      "step": 3990
+    },
+    {
+      "grad_norm": 0.5727388858795166,
+      "learning_rate": 7.00999121697069e-05,
+      "loss": 0.0435,
+      "step": 4000
+    },
+    {
+      "grad_norm": 0.441378653049469,
+      "learning_rate": 6.99484041580794e-05,
+      "loss": 0.0406,
+      "step": 4010
+    },
+    {
+      "grad_norm": 0.4964316487312317,
+      "learning_rate": 6.979667799390004e-05,
+      "loss": 0.0375,
+      "step": 4020
+    },
+    {
+      "grad_norm": 0.6407474279403687,
+      "learning_rate": 6.964473533642185e-05,
+      "loss": 0.0434,
+      "step": 4030
+    },
+    {
+      "grad_norm": 0.48845717310905457,
+      "learning_rate": 6.949257784726539e-05,
+      "loss": 0.0449,
+      "step": 4040
+    },
+    {
+      "grad_norm": 0.43027928471565247,
+      "learning_rate": 6.934020719040056e-05,
+      "loss": 0.0432,
+      "step": 4050
+    },
+    {
+      "grad_norm": 0.45439186692237854,
+      "learning_rate": 6.918762503212848e-05,
+      "loss": 0.0471,
+      "step": 4060
+    },
+    {
+      "grad_norm": 0.5205041766166687,
+      "learning_rate": 6.903483304106319e-05,
+      "loss": 0.0424,
+      "step": 4070
+    },
+    {
+      "grad_norm": 0.46493685245513916,
+      "learning_rate": 6.888183288811341e-05,
+      "loss": 0.0445,
+      "step": 4080
+    },
+    {
+      "grad_norm": 0.5404139161109924,
+      "learning_rate": 6.87286262464643e-05,
+      "loss": 0.0472,
+      "step": 4090
+    },
+    {
+      "grad_norm": 0.7282142043113708,
+      "learning_rate": 6.857521479155915e-05,
+      "loss": 0.0529,
+      "step": 4100
+    },
+    {
+      "grad_norm": 0.5229685306549072,
+      "learning_rate": 6.842160020108104e-05,
+      "loss": 0.0471,
+      "step": 4110
+    },
+    {
+      "grad_norm": 0.5389089584350586,
+      "learning_rate": 6.826778415493455e-05,
+      "loss": 0.0463,
+      "step": 4120
+    },
+    {
+      "grad_norm": 0.6004208326339722,
+      "learning_rate": 6.811376833522729e-05,
+      "loss": 0.0466,
+      "step": 4130
+    },
+    {
+      "grad_norm": 0.5835024118423462,
+      "learning_rate": 6.795955442625159e-05,
+      "loss": 0.048,
+      "step": 4140
+    },
+    {
+      "grad_norm": 0.46858230233192444,
+      "learning_rate": 6.780514411446608e-05,
+      "loss": 0.0468,
+      "step": 4150
+    },
+    {
+      "grad_norm": 0.5222543478012085,
+      "learning_rate": 6.765053908847716e-05,
+      "loss": 0.0431,
+      "step": 4160
+    },
+    {
+      "grad_norm": 0.49907833337783813,
+      "learning_rate": 6.749574103902064e-05,
+      "loss": 0.0511,
+      "step": 4170
+    },
+    {
+      "grad_norm": 0.45680519938468933,
+      "learning_rate": 6.734075165894317e-05,
+      "loss": 0.0465,
+      "step": 4180
+    },
+    {
+      "grad_norm": 0.4963977336883545,
+      "learning_rate": 6.71855726431838e-05,
+      "loss": 0.0445,
+      "step": 4190
+    },
+    {
+      "grad_norm": 0.5601822733879089,
+      "learning_rate": 6.703020568875538e-05,
+      "loss": 0.0411,
+      "step": 4200
+    },
+    {
+      "grad_norm": 0.4861275851726532,
+      "learning_rate": 6.687465249472603e-05,
+      "loss": 0.0378,
+      "step": 4210
+    },
+    {
+      "grad_norm": 0.4606975317001343,
+      "learning_rate": 6.671891476220055e-05,
+      "loss": 0.0376,
+      "step": 4220
+    },
+    {
+      "grad_norm": 0.43658024072647095,
+      "learning_rate": 6.656299419430183e-05,
+      "loss": 0.0404,
+      "step": 4230
+    },
+    {
+      "grad_norm": 0.45193716883659363,
+      "learning_rate": 6.640689249615223e-05,
+      "loss": 0.0447,
+      "step": 4240
+    },
+    {
+      "grad_norm": 0.5214183330535889,
+      "learning_rate": 6.625061137485491e-05,
+      "loss": 0.0441,
+      "step": 4250
+    },
+    {
+      "grad_norm": 0.5623442530632019,
+      "learning_rate": 6.609415253947517e-05,
+      "loss": 0.0485,
+      "step": 4260
+    },
+    {
+      "grad_norm": 0.5271878838539124,
+      "learning_rate": 6.593751770102178e-05,
+      "loss": 0.0425,
+      "step": 4270
+    },
+    {
+      "grad_norm": 0.40975555777549744,
+      "learning_rate": 6.578070857242823e-05,
+      "loss": 0.0397,
+      "step": 4280
+    },
+    {
+      "grad_norm": 0.42228594422340393,
+      "learning_rate": 6.562372686853402e-05,
+      "loss": 0.0378,
+      "step": 4290
+    },
+    {
+      "grad_norm": 0.45117226243019104,
+      "learning_rate": 6.546657430606593e-05,
+      "loss": 0.0411,
+      "step": 4300
+    },
+    {
+      "grad_norm": 0.5766953229904175,
+      "learning_rate": 6.530925260361918e-05,
+      "loss": 0.0376,
+      "step": 4310
+    },
+    {
+      "grad_norm": 0.4239300489425659,
+      "learning_rate": 6.515176348163871e-05,
+      "loss": 0.0426,
+      "step": 4320
+    },
+    {
+      "grad_norm": 0.4498891234397888,
+      "learning_rate": 6.499410866240032e-05,
+      "loss": 0.0387,
+      "step": 4330
+    },
+    {
+      "grad_norm": 0.5311325192451477,
+      "learning_rate": 6.48362898699919e-05,
+      "loss": 0.0381,
+      "step": 4340
+    },
+    {
+      "grad_norm": 0.5624609589576721,
+      "learning_rate": 6.467830883029443e-05,
+      "loss": 0.0414,
+      "step": 4350
+    },
+    {
+      "grad_norm": 0.4543378949165344,
+      "learning_rate": 6.452016727096326e-05,
+      "loss": 0.0396,
+      "step": 4360
+    },
+    {
+      "grad_norm": 0.47949710488319397,
+      "learning_rate": 6.436186692140916e-05,
+      "loss": 0.0398,
+      "step": 4370
+    },
+    {
+      "grad_norm": 0.43317073583602905,
+      "learning_rate": 6.420340951277938e-05,
+      "loss": 0.0391,
+      "step": 4380
+    },
+    {
+      "grad_norm": 0.3942078649997711,
+      "learning_rate": 6.404479677793874e-05,
+      "loss": 0.0415,
+      "step": 4390
+    },
+    {
+      "grad_norm": 0.6256000399589539,
+      "learning_rate": 6.388603045145075e-05,
+      "loss": 0.0422,
+      "step": 4400
+    },
+    {
+      "grad_norm": 0.4738806188106537,
+      "learning_rate": 6.372711226955843e-05,
+      "loss": 0.039,
+      "step": 4410
+    },
+    {
+      "grad_norm": 0.5231910943984985,
+      "learning_rate": 6.356804397016564e-05,
+      "loss": 0.0441,
+      "step": 4420
+    },
+    {
+      "grad_norm": 0.3898215889930725,
+      "learning_rate": 6.340882729281779e-05,
+      "loss": 0.0416,
+      "step": 4430
+    },
+    {
+      "grad_norm": 0.4737550914287567,
+      "learning_rate": 6.324946397868294e-05,
+      "loss": 0.0427,
+      "step": 4440
+    },
+    {
+      "grad_norm": 0.39828068017959595,
+      "learning_rate": 6.308995577053276e-05,
+      "loss": 0.0405,
+      "step": 4450
+    },
+    {
+      "grad_norm": 0.4809075593948364,
+      "learning_rate": 6.293030441272347e-05,
+      "loss": 0.0389,
+      "step": 4460
+    },
+    {
+      "grad_norm": 0.47399774193763733,
+      "learning_rate": 6.277051165117677e-05,
+      "loss": 0.0371,
+      "step": 4470
+    },
+    {
+      "grad_norm": 0.3674568831920624,
+      "learning_rate": 6.261057923336064e-05,
+      "loss": 0.0357,
+      "step": 4480
+    },
+    {
+      "grad_norm": 0.4679233729839325,
+      "learning_rate": 6.245050890827042e-05,
+      "loss": 0.0412,
+      "step": 4490
+    },
+    {
+      "grad_norm": 0.5382930636405945,
+      "learning_rate": 6.229030242640952e-05,
+      "loss": 0.0402,
+      "step": 4500
+    },
+    {
+      "grad_norm": 0.5336787104606628,
+      "learning_rate": 6.212996153977037e-05,
+      "loss": 0.0474,
+      "step": 4510
+    },
+    {
+      "grad_norm": 0.4890884459018707,
+      "learning_rate": 6.196948800181523e-05,
+      "loss": 0.0414,
+      "step": 4520
+    },
+    {
+      "grad_norm": 0.514552652835846,
+      "learning_rate": 6.180888356745695e-05,
+      "loss": 0.0398,
+      "step": 4530
+    },
+    {
+      "grad_norm": 0.5507141947746277,
+      "learning_rate": 6.164814999303995e-05,
+      "loss": 0.0414,
+      "step": 4540
+    },
+    {
+      "grad_norm": 0.45374220609664917,
+      "learning_rate": 6.148728903632081e-05,
+      "loss": 0.0357,
+      "step": 4550
+    },
+    {
+      "grad_norm": 0.5390177965164185,
+      "learning_rate": 6.132630245644921e-05,
+      "loss": 0.0367,
+      "step": 4560
+    },
+    {
+      "grad_norm": 0.4112001955509186,
+      "learning_rate": 6.116519201394857e-05,
+      "loss": 0.0389,
+      "step": 4570
+    },
+    {
+      "grad_norm": 0.3978293836116791,
+      "learning_rate": 6.10039594706969e-05,
+      "loss": 0.0378,
+      "step": 4580
+    },
+    {
+      "grad_norm": 0.34420955181121826,
+      "learning_rate": 6.084260658990744e-05,
+      "loss": 0.0358,
+      "step": 4590
+    },
+    {
+      "grad_norm": 0.47936877608299255,
+      "learning_rate": 6.068113513610943e-05,
+      "loss": 0.0363,
+      "step": 4600
+    },
+    {
+      "grad_norm": 0.3962310254573822,
+      "learning_rate": 6.0519546875128876e-05,
+      "loss": 0.0373,
+      "step": 4610
+    },
+    {
+      "grad_norm": 0.4856483042240143,
+      "learning_rate": 6.035784357406906e-05,
+      "loss": 0.0412,
+      "step": 4620
+    },
+    {
+      "grad_norm": 0.5573654174804688,
+      "learning_rate": 6.01960270012914e-05,
+      "loss": 0.0392,
+      "step": 4630
+    },
+    {
+      "grad_norm": 0.4923935532569885,
+      "learning_rate": 6.003409892639599e-05,
+      "loss": 0.0425,
+      "step": 4640
+    },
+    {
+      "grad_norm": 0.5231345891952515,
+      "learning_rate": 5.9872061120202336e-05,
+      "loss": 0.0427,
+      "step": 4650
+    },
+    {
+      "grad_norm": 0.43647822737693787,
+      "learning_rate": 5.9709915354729914e-05,
+      "loss": 0.0489,
+      "step": 4660
+    },
+    {
+      "grad_norm": 0.5012542605400085,
+      "learning_rate": 5.9547663403178824e-05,
+      "loss": 0.038,
+      "step": 4670
+    },
+    {
+      "grad_norm": 0.4920448362827301,
+      "learning_rate": 5.9385307039910445e-05,
+      "loss": 0.0378,
+      "step": 4680
+    },
+    {
+      "grad_norm": 0.5307269096374512,
+      "learning_rate": 5.922284804042792e-05,
+      "loss": 0.0446,
+      "step": 4690
+    },
+    {
+      "grad_norm": 0.5138248801231384,
+      "learning_rate": 5.906028818135687e-05,
+      "loss": 0.0468,
+      "step": 4700
+    },
+    {
+      "grad_norm": 0.4809703230857849,
+      "learning_rate": 5.889762924042585e-05,
+      "loss": 0.0375,
+      "step": 4710
+    },
+    {
+      "grad_norm": 0.446063756942749,
+      "learning_rate": 5.873487299644699e-05,
+      "loss": 0.0388,
+      "step": 4720
+    },
+    {
+      "grad_norm": 0.41934701800346375,
+      "learning_rate": 5.857202122929649e-05,
+      "loss": 0.0397,
+      "step": 4730
+    },
+    {
+      "grad_norm": 0.44001904129981995,
+      "learning_rate": 5.840907571989518e-05,
+      "loss": 0.0364,
+      "step": 4740
+    },
+    {
+      "grad_norm": 0.5035350322723389,
+      "learning_rate": 5.824603825018904e-05,
+      "loss": 0.0399,
+      "step": 4750
+    },
+    {
+      "grad_norm": 0.46448999643325806,
+      "learning_rate": 5.808291060312975e-05,
+      "loss": 0.0385,
+      "step": 4760
+    },
+    {
+      "grad_norm": 0.47263339161872864,
+      "learning_rate": 5.7919694562655083e-05,
+      "loss": 0.0389,
+      "step": 4770
+    },
+    {
+      "grad_norm": 0.44546037912368774,
+      "learning_rate": 5.775639191366954e-05,
+      "loss": 0.041,
+      "step": 4780
+    },
+    {
+      "grad_norm": 0.41190215945243835,
+      "learning_rate": 5.75930044420247e-05,
+      "loss": 0.0382,
+      "step": 4790
+    },
+    {
+      "grad_norm": 0.4584035277366638,
+      "learning_rate": 5.74295339344998e-05,
+      "loss": 0.0382,
+      "step": 4800
+    },
+    {
+      "grad_norm": 0.5399923324584961,
+      "learning_rate": 5.726598217878211e-05,
+      "loss": 0.0361,
+      "step": 4810
+    },
+    {
+      "grad_norm": 0.43452540040016174,
+      "learning_rate": 5.71023509634474e-05,
+      "loss": 0.0338,
+      "step": 4820
+    },
+    {
+      "grad_norm": 0.564422070980072,
+      "learning_rate": 5.693864207794049e-05,
+      "loss": 0.0395,
+      "step": 4830
+    },
+    {
+      "grad_norm": 0.37158694863319397,
+      "learning_rate": 5.677485731255545e-05,
+      "loss": 0.0467,
+      "step": 4840
+    },
+    {
+      "grad_norm": 0.4370258152484894,
+      "learning_rate": 5.6610998458416296e-05,
+      "loss": 0.0414,
+      "step": 4850
+    },
+    {
+      "grad_norm": 0.482177197933197,
+      "learning_rate": 5.644706730745716e-05,
+      "loss": 0.0404,
+      "step": 4860
+    },
+    {
+      "grad_norm": 0.38809412717819214,
+      "learning_rate": 5.628306565240287e-05,
+      "loss": 0.0393,
+      "step": 4870
+    },
+    {
+      "grad_norm": 0.38672083616256714,
+      "learning_rate": 5.611899528674923e-05,
+      "loss": 0.0446,
+      "step": 4880
+    },
+    {
+      "grad_norm": 0.4755399525165558,
+      "learning_rate": 5.595485800474349e-05,
+      "loss": 0.042,
+      "step": 4890
+    },
+    {
+      "grad_norm": 0.49957188963890076,
+      "learning_rate": 5.579065560136467e-05,
+      "loss": 0.0398,
+      "step": 4900
+    },
+    {
+      "grad_norm": 0.3985891342163086,
+      "learning_rate": 5.562638987230392e-05,
+      "loss": 0.0389,
+      "step": 4910
+    },
+    {
+      "grad_norm": 0.3723376393318176,
+      "learning_rate": 5.546206261394498e-05,
+      "loss": 0.0363,
+      "step": 4920
+    },
+    {
+      "grad_norm": 0.49992597103118896,
+      "learning_rate": 5.529767562334437e-05,
+      "loss": 0.0411,
+      "step": 4930
+    },
+    {
+      "grad_norm": 0.5506955981254578,
+      "learning_rate": 5.5133230698211926e-05,
+      "loss": 0.0383,
+      "step": 4940
+    },
+    {
+      "grad_norm": 0.4882294833660126,
+      "learning_rate": 5.496872963689096e-05,
+      "loss": 0.0382,
+      "step": 4950
+    },
+    {
+      "grad_norm": 0.4228740334510803,
+      "learning_rate": 5.4804174238338756e-05,
+      "loss": 0.034,
+      "step": 4960
+    },
+    {
+      "grad_norm": 0.5484710335731506,
+      "learning_rate": 5.463956630210678e-05,
+      "loss": 0.0463,
+      "step": 4970
+    },
+    {
+      "grad_norm": 0.37834620475769043,
+      "learning_rate": 5.4474907628321046e-05,
+      "loss": 0.0415,
+      "step": 4980
+    },
+    {
+      "grad_norm": 0.48790180683135986,
+      "learning_rate": 5.431020001766244e-05,
+      "loss": 0.0394,
+      "step": 4990
+    },
+    {
+      "grad_norm": 0.5073119401931763,
+      "learning_rate": 5.4145445271346986e-05,
+      "loss": 0.0373,
+      "step": 5000
+    },
+    {
+      "grad_norm": 0.5300912261009216,
+      "learning_rate": 5.398064519110622e-05,
+      "loss": 0.0373,
+      "step": 5010
+    },
+    {
+      "grad_norm": 0.4546026885509491,
+      "learning_rate": 5.3815801579167394e-05,
+      "loss": 0.0362,
+      "step": 5020
+    },
+    {
+      "grad_norm": 0.4915795922279358,
+      "learning_rate": 5.365091623823382e-05,
+      "loss": 0.0367,
+      "step": 5030
+    },
+    {
+      "grad_norm": 0.42918261885643005,
+      "learning_rate": 5.348599097146521e-05,
+      "loss": 0.0352,
+      "step": 5040
+    },
+    {
+      "grad_norm": 0.4628450274467468,
+      "learning_rate": 5.3321027582457836e-05,
+      "loss": 0.0374,
+      "step": 5050
+    },
+    {
+      "grad_norm": 0.5019720196723938,
+      "learning_rate": 5.315602787522491e-05,
+      "loss": 0.036,
+      "step": 5060
+    },
+    {
+      "grad_norm": 0.4162551164627075,
+      "learning_rate": 5.299099365417678e-05,
+      "loss": 0.0335,
+      "step": 5070
+    },
+    {
+      "grad_norm": 0.44460800290107727,
+      "learning_rate": 5.2825926724101236e-05,
+      "loss": 0.0397,
+      "step": 5080
+    },
+    {
+      "grad_norm": 0.36362424492836,
+      "learning_rate": 5.26608288901438e-05,
+      "loss": 0.0382,
+      "step": 5090
+    },
+    {
+      "grad_norm": 0.44403377175331116,
+      "learning_rate": 5.24957019577879e-05,
+      "loss": 0.0389,
+      "step": 5100
+    },
+    {
+      "grad_norm": 0.4362407624721527,
+      "learning_rate": 5.2330547732835266e-05,
+      "loss": 0.0352,
+      "step": 5110
+    },
+    {
+      "grad_norm": 0.5165354609489441,
+      "learning_rate": 5.2165368021385996e-05,
+      "loss": 0.0453,
+      "step": 5120
+    },
+    {
+      "grad_norm": 0.5299188494682312,
+      "learning_rate": 5.200016462981897e-05,
+      "loss": 0.0345,
+      "step": 5130
+    },
+    {
+      "grad_norm": 0.5661501288414001,
+      "learning_rate": 5.1834939364772015e-05,
+      "loss": 0.0381,
+      "step": 5140
+    },
+    {
+      "grad_norm": 0.5396561026573181,
+      "learning_rate": 5.166969403312214e-05,
+      "loss": 0.0344,
+      "step": 5150
+    },
+    {
+      "grad_norm": 0.4610588550567627,
+      "learning_rate": 5.1504430441965844e-05,
+      "loss": 0.0386,
+      "step": 5160
+    },
+    {
+      "grad_norm": 0.4185340404510498,
+      "learning_rate": 5.133915039859923e-05,
+      "loss": 0.0321,
+      "step": 5170
+    },
+    {
+      "grad_norm": 0.5117337107658386,
+      "learning_rate": 5.1173855710498444e-05,
+      "loss": 0.0363,
+      "step": 5180
+    },
+    {
+      "grad_norm": 0.4781377613544464,
+      "learning_rate": 5.100854818529967e-05,
+      "loss": 0.036,
+      "step": 5190
+    },
+    {
+      "grad_norm": 0.4905393421649933,
+      "learning_rate": 5.084322963077951e-05,
+      "loss": 0.0407,
+      "step": 5200
+    },
+    {
+      "grad_norm": 0.4550723731517792,
+      "learning_rate": 5.067790185483522e-05,
+      "loss": 0.0379,
+      "step": 5210
+    },
+    {
+      "grad_norm": 0.45524081587791443,
+      "learning_rate": 5.0512566665464844e-05,
+      "loss": 0.0383,
+      "step": 5220
+    },
+    {
+      "grad_norm": 0.4141499996185303,
+      "learning_rate": 5.034722587074755e-05,
+      "loss": 0.0359,
+      "step": 5230
+    },
+    {
+      "grad_norm": 0.4905252158641815,
+      "learning_rate": 5.018188127882375e-05,
+      "loss": 0.0355,
+      "step": 5240
+    },
+    {
+      "grad_norm": 0.42627424001693726,
+      "learning_rate": 5.0016534697875417e-05,
+      "loss": 0.0328,
+      "step": 5250
+    },
+    {
+      "grad_norm": 0.38990166783332825,
+      "learning_rate": 4.9851187936106294e-05,
+      "loss": 0.0373,
+      "step": 5260
+    },
+    {
+      "grad_norm": 0.41246482729911804,
+      "learning_rate": 4.968584280172206e-05,
+      "loss": 0.0359,
+      "step": 5270
+    },
+    {
+      "grad_norm": 0.4924166202545166,
+      "learning_rate": 4.95205011029106e-05,
+      "loss": 0.0339,
+      "step": 5280
+    },
+    {
+      "grad_norm": 0.37498942017555237,
+      "learning_rate": 4.935516464782227e-05,
+      "loss": 0.036,
+      "step": 5290
+    },
+    {
+      "grad_norm": 0.3869049549102783,
+      "learning_rate": 4.918983524455003e-05,
+      "loss": 0.0347,
+      "step": 5300
+    },
+    {
+      "grad_norm": 0.47202593088150024,
+      "learning_rate": 4.9024514701109766e-05,
+      "loss": 0.0324,
+      "step": 5310
+    },
+    {
+      "grad_norm": 0.4898712635040283,
+      "learning_rate": 4.885920482542043e-05,
+      "loss": 0.0353,
+      "step": 5320
+    },
+    {
+      "grad_norm": 0.395417720079422,
+      "learning_rate": 4.869390742528438e-05,
+      "loss": 0.0337,
+      "step": 5330
+    },
+    {
+      "grad_norm": 0.4707302153110504,
+      "learning_rate": 4.852862430836744e-05,
+      "loss": 0.0377,
+      "step": 5340
+    },
+    {
+      "grad_norm": 0.42319411039352417,
+      "learning_rate": 4.836335728217933e-05,
+      "loss": 0.0351,
+      "step": 5350
+    },
+    {
+      "grad_norm": 0.4861985146999359,
+      "learning_rate": 4.819810815405379e-05,
+      "loss": 0.0379,
+      "step": 5360
+    },
+    {
+      "grad_norm": 0.6579845547676086,
+      "learning_rate": 4.803287873112877e-05,
+      "loss": 0.0391,
+      "step": 5370
+    },
+    {
+      "grad_norm": 0.4598766267299652,
+      "learning_rate": 4.786767082032681e-05,
+      "loss": 0.0373,
+      "step": 5380
+    },
+    {
+      "grad_norm": 0.4411160349845886,
+      "learning_rate": 4.77024862283351e-05,
+      "loss": 0.0333,
+      "step": 5390
+    },
+    {
+      "grad_norm": 0.3847849369049072,
+      "learning_rate": 4.753732676158593e-05,
+      "loss": 0.0317,
+      "step": 5400
+    },
+    {
+      "grad_norm": 0.49002495408058167,
+      "learning_rate": 4.737219422623672e-05,
+      "loss": 0.0364,
+      "step": 5410
+    },
+    {
+      "grad_norm": 0.39545437693595886,
+      "learning_rate": 4.720709042815044e-05,
+      "loss": 0.0321,
+      "step": 5420
+    },
+    {
+      "grad_norm": 0.46968668699264526,
+      "learning_rate": 4.704201717287578e-05,
+      "loss": 0.033,
+      "step": 5430
+    },
+    {
+      "grad_norm": 0.4025862514972687,
+      "learning_rate": 4.6876976265627404e-05,
+      "loss": 0.036,
+      "step": 5440
+    },
+    {
+      "grad_norm": 0.6391124725341797,
+      "learning_rate": 4.671196951126626e-05,
+      "loss": 0.0382,
+      "step": 5450
+    },
+    {
+      "grad_norm": 0.5088096857070923,
+      "learning_rate": 4.654699871427971e-05,
+      "loss": 0.0321,
+      "step": 5460
+    },
+    {
+      "grad_norm": 0.43340057134628296,
+      "learning_rate": 4.6382065678762034e-05,
+      "loss": 0.0347,
+      "step": 5470
+    },
+    {
+      "grad_norm": 0.413905531167984,
+      "learning_rate": 4.6217172208394424e-05,
+      "loss": 0.0351,
+      "step": 5480
+    },
+    {
+      "grad_norm": 0.3292362689971924,
+      "learning_rate": 4.605232010642549e-05,
+      "loss": 0.0315,
+      "step": 5490
+    },
+    {
+      "grad_norm": 0.5179662704467773,
+      "learning_rate": 4.588751117565142e-05,
+      "loss": 0.0321,
+      "step": 5500
+    },
+    {
+      "grad_norm": 0.39133188128471375,
+      "learning_rate": 4.5722747218396214e-05,
+      "loss": 0.0341,
+      "step": 5510
+    },
+    {
+      "grad_norm": 0.5115916132926941,
+      "learning_rate": 4.5558030036492194e-05,
+      "loss": 0.0328,
+      "step": 5520
+    },
+    {
+      "grad_norm": 0.38737472891807556,
+      "learning_rate": 4.539336143125999e-05,
+      "loss": 0.0405,
+      "step": 5530
+    },
+    {
+      "grad_norm": 0.4076855480670929,
+      "learning_rate": 4.522874320348916e-05,
+      "loss": 0.0331,
+      "step": 5540
+    },
+    {
+      "grad_norm": 0.502906858921051,
+      "learning_rate": 4.506417715341821e-05,
+      "loss": 0.0313,
+      "step": 5550
+    },
+    {
+      "grad_norm": 0.4934130012989044,
+      "learning_rate": 4.489966508071511e-05,
+      "loss": 0.0329,
+      "step": 5560
+    },
+    {
+      "grad_norm": 0.4525161385536194,
+      "learning_rate": 4.4735208784457575e-05,
+      "loss": 0.0333,
+      "step": 5570
+    },
+    {
+      "grad_norm": 0.44312161207199097,
+      "learning_rate": 4.457081006311325e-05,
+      "loss": 0.0343,
+      "step": 5580
+    },
+    {
+      "grad_norm": 0.45065921545028687,
+      "learning_rate": 4.440647071452027e-05,
+      "loss": 0.0363,
+      "step": 5590
+    },
+    {
+      "grad_norm": 0.3948800265789032,
+      "learning_rate": 4.424219253586737e-05,
+      "loss": 0.0371,
+      "step": 5600
+    },
+    {
+      "grad_norm": 0.42451366782188416,
+      "learning_rate": 4.407797732367443e-05,
+      "loss": 0.0349,
+      "step": 5610
+    },
+    {
+      "grad_norm": 0.4344401955604553,
+      "learning_rate": 4.391382687377268e-05,
+      "loss": 0.0344,
+      "step": 5620
+    },
+    {
+      "grad_norm": 0.43276187777519226,
+      "learning_rate": 4.374974298128512e-05,
+      "loss": 0.0322,
+      "step": 5630
+    },
+    {
+      "grad_norm": 0.3546721637248993,
+      "learning_rate": 4.358572744060699e-05,
+      "loss": 0.0407,
+      "step": 5640
+    },
+    {
+      "grad_norm": 0.3919878304004669,
+      "learning_rate": 4.342178204538588e-05,
+      "loss": 0.0279,
+      "step": 5650
+    },
+    {
+      "grad_norm": 0.4614352881908417,
+      "learning_rate": 4.325790858850241e-05,
+      "loss": 0.0359,
+      "step": 5660
+    },
+    {
+      "grad_norm": 0.4700967073440552,
+      "learning_rate": 4.309410886205043e-05,
+      "loss": 0.0305,
+      "step": 5670
+    },
+    {
+      "grad_norm": 0.4928377866744995,
+      "learning_rate": 4.293038465731752e-05,
+      "loss": 0.0321,
+      "step": 5680
+    },
+    {
+      "grad_norm": 0.4820231795310974,
+      "learning_rate": 4.276673776476533e-05,
+      "loss": 0.0367,
+      "step": 5690
+    },
+    {
+      "grad_norm": 0.31659793853759766,
+      "learning_rate": 4.260316997401007e-05,
+      "loss": 0.0301,
+      "step": 5700
+    },
+    {
+      "grad_norm": 0.4053920805454254,
+      "learning_rate": 4.243968307380293e-05,
+      "loss": 0.0336,
+      "step": 5710
+    },
+    {
+      "grad_norm": 0.48853442072868347,
+      "learning_rate": 4.22762788520104e-05,
+      "loss": 0.0364,
+      "step": 5720
+    },
+    {
+      "grad_norm": 0.4464189410209656,
+      "learning_rate": 4.211295909559491e-05,
+      "loss": 0.038,
+      "step": 5730
+    },
+    {
+      "grad_norm": 0.4845777153968811,
+      "learning_rate": 4.194972559059511e-05,
+      "loss": 0.0315,
+      "step": 5740
+    },
+    {
+      "grad_norm": 0.3993183970451355,
+      "learning_rate": 4.178658012210651e-05,
+      "loss": 0.0308,
+      "step": 5750
+    },
+    {
+      "grad_norm": 0.4467678368091583,
+      "learning_rate": 4.162352447426177e-05,
+      "loss": 0.0333,
+      "step": 5760
+    },
+    {
+      "grad_norm": 0.4443342685699463,
+      "learning_rate": 4.146056043021135e-05,
+      "loss": 0.0322,
+      "step": 5770
+    },
+    {
+      "grad_norm": 0.35077914595603943,
+      "learning_rate": 4.1297689772103944e-05,
+      "loss": 0.0333,
+      "step": 5780
+    },
+    {
+      "grad_norm": 0.4200893044471741,
+      "learning_rate": 4.113491428106694e-05,
+      "loss": 0.0326,
+      "step": 5790
+    },
+    {
+      "grad_norm": 0.4066666066646576,
+      "learning_rate": 4.0972235737187055e-05,
+      "loss": 0.0311,
+      "step": 5800
+    },
+    {
+      "grad_norm": 0.342128187417984,
+      "learning_rate": 4.080965591949076e-05,
+      "loss": 0.0304,
+      "step": 5810
+    },
+    {
+      "grad_norm": 0.4631299078464508,
+      "learning_rate": 4.0647176605924924e-05,
+      "loss": 0.0312,
+      "step": 5820
+    },
+    {
+      "grad_norm": 0.5019406080245972,
+      "learning_rate": 4.0484799573337255e-05,
+      "loss": 0.0342,
+      "step": 5830
+    },
+    {
+      "grad_norm": 0.5158159732818604,
+      "learning_rate": 4.032252659745699e-05,
+      "loss": 0.0329,
+      "step": 5840
+    },
+    {
+      "grad_norm": 0.5040042400360107,
+      "learning_rate": 4.016035945287539e-05,
+      "loss": 0.0412,
+      "step": 5850
+    },
+    {
+      "grad_norm": 0.4785839319229126,
+      "learning_rate": 3.999829991302635e-05,
+      "loss": 0.0343,
+      "step": 5860
+    },
+    {
+      "grad_norm": 0.4362504780292511,
+      "learning_rate": 3.983634975016707e-05,
+      "loss": 0.0306,
+      "step": 5870
+    },
+    {
+      "grad_norm": 0.4862264096736908,
+      "learning_rate": 3.967451073535854e-05,
+      "loss": 0.0418,
+      "step": 5880
+    },
+    {
+      "grad_norm": 0.43010035157203674,
+      "learning_rate": 3.951278463844633e-05,
+      "loss": 0.0302,
+      "step": 5890
+    },
+    {
+      "grad_norm": 0.5377092361450195,
+      "learning_rate": 3.935117322804111e-05,
+      "loss": 0.0363,
+      "step": 5900
+    },
+    {
+      "grad_norm": 0.40336716175079346,
+      "learning_rate": 3.918967827149938e-05,
+      "loss": 0.0348,
+      "step": 5910
+    },
+    {
+      "grad_norm": 0.4090577960014343,
+      "learning_rate": 3.9028301534904094e-05,
+      "loss": 0.0354,
+      "step": 5920
+    },
+    {
+      "grad_norm": 0.36200812458992004,
+      "learning_rate": 3.88670447830454e-05,
+      "loss": 0.0304,
+      "step": 5930
+    },
+    {
+      "grad_norm": 0.3312746286392212,
+      "learning_rate": 3.870590977940132e-05,
+      "loss": 0.0316,
+      "step": 5940
+    },
+    {
+      "grad_norm": 0.3790975511074066,
+      "learning_rate": 3.8544898286118404e-05,
+      "loss": 0.0295,
+      "step": 5950
+    },
+    {
+      "grad_norm": 0.44410622119903564,
+      "learning_rate": 3.838401206399257e-05,
+      "loss": 0.0297,
+      "step": 5960
+    },
+    {
+      "grad_norm": 0.38737359642982483,
+      "learning_rate": 3.822325287244975e-05,
+      "loss": 0.0303,
+      "step": 5970
+    },
+    {
+      "grad_norm": 0.5516029596328735,
+      "learning_rate": 3.8062622469526725e-05,
+      "loss": 0.0333,
+      "step": 5980
+    },
+    {
+      "grad_norm": 0.39691001176834106,
+      "learning_rate": 3.790212261185183e-05,
+      "loss": 0.0324,
+      "step": 5990
+    },
+    {
+      "grad_norm": 0.44088539481163025,
+      "learning_rate": 3.7741755054625794e-05,
+      "loss": 0.0315,
+      "step": 6000
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 10000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 3000,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

gr00t/80_20_gr00t/checkpoint-6000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d223ff78a7bd2f2dddd363e2f1515e00aaabad284080b6e91ca0664ba1755bd5
+size 5841

gr00t/80_20_gr00t/checkpoint-6000/wandb_config.json

@@ -0,0 +1 @@
+{"project": "finetune-gr00t-n1d6", "run_id": "80_20_gr00t"}

gr00t/80_20_gr00t/checkpoint-9000/config.json

@@ -0,0 +1,70 @@
+{
+  "action_horizon": 50,
+  "add_pos_embed": true,
+  "apply_sincos_state_encoding": true,
+  "architectures": [
+    "Gr00tN1d6"
+  ],
+  "attn_dropout": 0.2,
+  "attn_implementation": null,
+  "backbone_embedding_dim": 2048,
+  "backbone_model_type": "eagle",
+  "backbone_trainable_params_fp32": true,
+  "collator_overwrite_image_inputs": false,
+  "color_jitter_params": {
+    "brightness": 0.1,
+    "contrast": 0.1,
+    "hue": 0.1,
+    "saturation": 0.1
+  },
+  "crop_fraction": 0.95,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "eagle_collator": true,
+  "formalize_language": true,
+  "gemma_collator": false,
+  "hidden_size": 1024,
+  "image_crop_size": null,
+  "image_target_size": null,
+  "input_embedding_dim": 1536,
+  "load_bf16": true,
+  "max_action_dim": 128,
+  "max_num_embodiments": 32,
+  "max_seq_len": 1024,
+  "max_state_dim": 128,
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "model_type": "Gr00tN1d6",
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_inference_timesteps": 4,
+  "num_timestep_buckets": 1000,
+  "random_rotation_angle": null,
+  "reproject_vision": false,
+  "select_layer": 16,
+  "shortest_image_edge": 256,
+  "state_dropout_prob": 0.0,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3",
+  "tune_diffusion_model": true,
+  "tune_llm": true,
+  "tune_projector": true,
+  "tune_top_llm_layers": 4,
+  "tune_visual": false,
+  "tune_vlln": true,
+  "use_albumentations_transforms": true,
+  "use_alternate_vl_dit": true,
+  "use_flash_attention": true,
+  "use_relative_action": true,
+  "use_vlln": true
+}

gr00t/80_20_gr00t/checkpoint-9000/embodiment_id.json

@@ -0,0 +1,11 @@
+{
+  "robocasa_panda_omron": 13,
+  "gr1": 20,
+  "behavior_r1_pro": 24,
+  "unitree_g1": 8,
+  "oxe_google": 0,
+  "oxe_widowx": 1,
+  "libero_panda": 2,
+  "oxe_droid": 16,
+  "new_embodiment": 10
+}

gr00t/80_20_gr00t/checkpoint-9000/experiment_cfg/conf.yaml

@@ -0,0 +1,246 @@
+load_config_path: null
+model:
+  model_type: Gr00tN1d6
+  model_dtype: bfloat16
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  backbone_model_type: eagle
+  model_revision: null
+  tune_top_llm_layers: 4
+  backbone_embedding_dim: 2048
+  tune_llm: true
+  tune_visual: false
+  select_layer: 16
+  reproject_vision: false
+  use_flash_attention: true
+  load_bf16: false
+  collator_overwrite_image_inputs: false
+  eagle_collator: true
+  backbone_trainable_params_fp32: true
+  image_crop_size: null
+  image_target_size: null
+  shortest_image_edge: 256
+  crop_fraction: 0.95
+  random_rotation_angle: null
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  use_albumentations_transforms: true
+  formalize_language: true
+  apply_sincos_state_encoding: false
+  use_relative_action: true
+  max_state_dim: 29
+  max_action_dim: 29
+  action_horizon: 16
+  hidden_size: 1024
+  input_embedding_dim: 1536
+  add_pos_embed: true
+  attn_dropout: 0.2
+  use_vlln: true
+  max_seq_len: 1024
+  use_alternate_vl_dit: true
+  attend_text_every_n_blocks: 2
+  diffusion_model_cfg:
+    positional_embeddings: null
+    num_layers: 32
+    num_attention_heads: 32
+    attention_head_dim: 48
+    norm_type: ada_norm
+    dropout: 0.2
+    final_dropout: true
+    output_dim: 1024
+    interleave_self_attention: true
+  num_inference_timesteps: 4
+  noise_beta_alpha: 1.5
+  noise_beta_beta: 1.0
+  noise_s: 0.999
+  num_timestep_buckets: 1000
+  tune_projector: true
+  tune_diffusion_model: true
+  tune_vlln: true
+  state_dropout_prob: 0.0
+  state_additive_noise_scale: 0.0
+  max_num_embodiments: 32
+data:
+  datasets:
+  - dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  modality_configs:
+    new_embodiment:
+      video:
+        delta_indices:
+        - 0
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      state:
+        delta_indices:
+        - 0
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      action:
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        modality_keys:
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs:
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+        - rep: ABSOLUTE
+          type: NON_EEF
+          format: DEFAULT
+          state_key: null
+      language:
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+  download_cache: false
+  shard_size: 1024
+  episode_sampling_rate: 0.1
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  mode: single_turn
+  random_chop: 0.0
+  mock_dataset_mode: false
+  shuffle: true
+  seed: 42
+  multiprocessing_context: fork
+  allow_padding: false
+  subsample_ratio: 1.0
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  video_backend: torchcodec
+training:
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  experiment_name: null
+  max_steps: 10000
+  global_batch_size: 32
+  batch_size: null
+  gradient_accumulation_steps: 1
+  learning_rate: 0.0001
+  lr_scheduler_type: cosine
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  max_grad_norm: 1.0
+  optim: adamw_torch
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  fp16: false
+  bf16: true
+  eval_bf16: true
+  logging_steps: 10
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  max_concurrent_uploads: 2
+  eval_strategy: 'no'
+  eval_steps: 500
+  eval_set_split_ratio: 0.1
+  eval_batch_size: 2
+  save_best_eval_metric_name: ''
+  save_best_eval_metric_greater_is_better: true
+  deepspeed_stage: 2
+  gradient_checkpointing: false
+  transformers_trust_remote_code: true
+  transformers_local_files_only: false
+  transformers_cache_dir: null
+  transformers_access_token: null
+  use_ddp: false
+  ddp_bucket_cap_mb: 100
+  num_gpus: 1
+  dataloader_num_workers: 4
+  remove_unused_columns: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  enable_profiling: false
+  max_retries: 3
+  assert_loss_less_than: null
+  add_rl_callback: false
+  enable_open_loop_eval: false
+  open_loop_eval_traj_ids:
+  - 0
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_plot_indices: null
+max_steps: 10000
+save_steps: 3000

gr00t/80_20_gr00t/checkpoint-9000/experiment_cfg/config.yaml

@@ -0,0 +1,277 @@
+!!python/object:gr00t.configs.base_config.Config
+data: !!python/object:gr00t.configs.data.data_config.DataConfig
+  allow_padding: false
+  datasets:
+  - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
+    dataset_paths:
+    - /workspace/capstone-vla/datasets/syn_real_mix/syn_realDataset/mixed_80_20
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 1.0
+    val_dataset_path: null
+  download_cache: false
+  episode_sampling_rate: 0.1
+  image_crop_size:
+  - 244
+  - 244
+  image_target_size:
+  - 224
+  - 224
+  mock_dataset_mode: false
+  modality_configs:
+    new_embodiment:
+      action: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs:
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
+          - default
+          rep: &id002 !!python/object/apply:gr00t.data.types.ActionRepresentation
+          - absolute
+          state_key: null
+          type: &id003 !!python/object/apply:gr00t.data.types.ActionType
+          - non_eef
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: *id002
+          state_key: null
+          type: *id003
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        mean_std_embedding_keys: null
+        modality_keys: &id004
+        - left_arm
+        - left_hand
+        - left_leg
+        - neck
+        - right_arm
+        - right_hand
+        - right_leg
+        - waist
+        sin_cos_embedding_keys: null
+      language: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - annotation.human.action.task_description
+        sin_cos_embedding_keys: null
+      state: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys: *id004
+        sin_cos_embedding_keys: null
+      video: !!python/object:gr00t.data.types.ModalityConfig
+        action_configs: null
+        delta_indices:
+        - 0
+        mean_std_embedding_keys: null
+        modality_keys:
+        - ego_view
+        - left_wrist_view
+        - right_wrist_view
+        sin_cos_embedding_keys: null
+  mode: single_turn
+  multiprocessing_context: fork
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: false
+  random_chop: 0.0
+  seed: 42
+  shard_size: 1024
+  shuffle: true
+  subsample_ratio: 1.0
+  video_backend: torchcodec
+load_config_path: null
+model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
+  _attn_implementation_autoset: false
+  _attn_implementation_internal: null
+  _commit_hash: null
+  _name_or_path: ''
+  add_cross_attention: false
+  architectures: null
+  backbone_model_type: eagle
+  backbone_trainable_params_fp32: true
+  bad_words_ids: null
+  begin_suppress_tokens: null
+  bos_token_id: null
+  chunk_size_feed_forward: 0
+  color_jitter_params:
+    brightness: 0.3
+    contrast: 0.4
+    hue: 0.08
+    saturation: 0.5
+  cross_attention_hidden_size: null
+  decoder_start_token_id: null
+  diffusion_model_cfg:
+    attention_head_dim: 48
+    dropout: 0.2
+    final_dropout: true
+    interleave_self_attention: true
+    norm_type: ada_norm
+    num_attention_heads: 32
+    num_layers: 32
+    output_dim: 1024
+    positional_embeddings: null
+  diversity_penalty: 0.0
+  do_sample: false
+  eagle_collator: true
+  early_stopping: false
+  encoder_no_repeat_ngram_size: 0
+  eos_token_id: null
+  exponential_decay_length_penalty: null
+  finetuning_task: null
+  forced_bos_token_id: null
+  forced_eos_token_id: null
+  id2label:
+    0: LABEL_0
+    1: LABEL_1
+  is_decoder: false
+  is_encoder_decoder: false
+  label2id:
+    LABEL_0: 0
+    LABEL_1: 1
+  length_penalty: 1.0
+  load_bf16: false
+  max_length: 20
+  min_length: 0
+  model_name: nvidia/Eagle-Block2A-2B-v2
+  no_repeat_ngram_size: 0
+  num_beam_groups: 1
+  num_beams: 1
+  num_return_sequences: 1
+  output_attentions: false
+  output_hidden_states: false
+  output_scores: false
+  pad_token_id: null
+  prefix: null
+  problem_type: null
+  pruned_heads: {}
+  random_rotation_angle: null
+  remove_invalid_values: false
+  repetition_penalty: 1.0
+  reproject_vision: false
+  return_dict: true
+  return_dict_in_generate: false
+  sep_token_id: null
+  state_dropout_prob: 0.0
+  suppress_tokens: null
+  task_specific_params: null
+  temperature: 1.0
+  tf_legacy_loss: false
+  tie_encoder_decoder: false
+  tie_word_embeddings: true
+  tokenizer_class: null
+  top_k: 50
+  top_p: 1.0
+  torch_dtype: null
+  torchscript: false
+  transformers_version: null
+  tune_diffusion_model: true
+  tune_llm: true
+  tune_projector: true
+  tune_visual: false
+  typical_p: 1.0
+  use_bfloat16: false
+  use_relative_action: true
+training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
+  add_rl_callback: false
+  assert_loss_less_than: null
+  batch_size: null
+  bf16: true
+  dataloader_num_workers: 4
+  ddp_bucket_cap_mb: 100
+  deepspeed_stage: 2
+  enable_open_loop_eval: false
+  enable_profiling: false
+  eval_batch_size: 2
+  eval_bf16: true
+  eval_set_split_ratio: 0.1
+  eval_steps: 500
+  eval_strategy: 'no'
+  experiment_name: null
+  fp16: false
+  global_batch_size: 32
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: false
+  learning_rate: 0.0001
+  logging_steps: 10
+  lr_scheduler_type: cosine
+  max_concurrent_uploads: 2
+  max_grad_norm: 1.0
+  max_retries: 3
+  max_steps: 10000
+  num_gpus: 1
+  open_loop_eval_plot_indices: null
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_traj_ids:
+  - 0
+  optim: adamw_torch
+  output_dir: /workspace/capstone-vla/third_party/gr00t/chkpt_rlwd/80_20_gr00t
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_steps: 3000
+  save_total_limit: 3
+  save_vl_model: false
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  transformers_access_token: null
+  transformers_cache_dir: null
+  transformers_local_files_only: false
+  transformers_trust_remote_code: true
+  upload_checkpoints: false
+  upload_every: 1000
+  upload_last_n_checkpoints: 5
+  use_ddp: false
+  use_wandb: true
+  wandb_project: finetune-gr00t-n1d6
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  weight_decay: 1.0e-05

gr00t/80_20_gr00t/checkpoint-9000/experiment_cfg/dataset_statistics.json

@@ -0,0 +1,761 @@
+{
+  "new_embodiment": {
+    "state": {
+      "left_arm": {
+        "min": [
+          -1.4764578342437744,
+          -0.1518283635377884,
+          -1.499962568283081,
+          -2.27626633644104,
+          -1.5671002864837646,
+          -0.6656755208969116,
+          -0.5207215547561646
+        ],
+        "max": [
+          0.3441833555698395,
+          0.657663881778717,
+          1.2758116722106934,
+          0.7662562131881714,
+          0.5798436999320984,
+          0.9854870438575745,
+          0.6866571307182312
+        ],
+        "mean": [
+          -0.24674749374389648,
+          0.27366140484809875,
+          -0.37971556186676025,
+          -1.63396155834198,
+          -0.1316976696252823,
+          0.2934497594833374,
+          -0.003799525322392583
+        ],
+        "std": [
+          0.2643950283527374,
+          0.06967416405677795,
+          0.5029074549674988,
+          0.685847282409668,
+          0.41868942975997925,
+          0.2417760193347931,
+          0.23273395001888275
+        ],
+        "q01": [
+          -1.1877377033233643,
+          0.13130172491073605,
+          -1.4007886266708374,
+          -2.166787986755371,
+          -1.5548944282531738,
+          -0.4001814353466034,
+          -0.34637925028800964
+        ],
+        "q99": [
+          0.10601000964642036,
+          0.46540972113609314,
+          1.1985569143295296,
+          0.6938967108726501,
+          0.2971799075603485,
+          0.9169320225715639,
+          0.5222405481338501
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7414746284484863,
+          -0.16363811492919922,
+          -0.741387128829956,
+          -0.6882719993591309,
+          -1.5692027807235718,
+          -1.5692027807235718
+        ],
+        "std": [
+          1.2994686365127563,
+          1.4938863515853882,
+          1.2995597124099731,
+          1.3310250043869019,
+          2.54817533493042,
+          2.54817533493042
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7119174599647522,
+          -0.55275958776474,
+          -0.09536845237016678,
+          -2.144679546356201,
+          -0.6652951240539551,
+          -0.7053864002227783,
+          -0.5874713063240051
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.433588981628418,
+          0.9927297830581665,
+          0.6599544286727905
+        ],
+        "mean": [
+          -0.1792619228363037,
+          -0.1758386492729187,
+          0.46474239230155945,
+          -1.5775669813156128,
+          0.06633679568767548,
+          -0.13522601127624512,
+          -0.0053453403525054455
+        ],
+        "std": [
+          0.2031964510679245,
+          0.17344926297664642,
+          0.2641066610813141,
+          0.44620281457901,
+          0.5178105235099792,
+          0.2680231034755707,
+          0.1985798329114914
+        ],
+        "q01": [
+          -0.5415496277809143,
+          -0.43221070885658264,
+          0.05797714069485664,
+          -2.007323226928711,
+          -0.5142281770706176,
+          -0.6404054927825927,
+          -0.3437119019031525
+        ],
+        "q99": [
+          0.2733417749404907,
+          0.4016014683246615,
+          1.2267546653747559,
+          0.6918689608573914,
+          1.3172417879104614,
+          0.8524364829063416,
+          0.4512056303024375
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40354546904563904,
+          0.11602523177862167,
+          -0.4034579396247864,
+          -0.4363398253917694,
+          -0.8933442831039429,
+          -0.8933442831039429
+        ],
+        "std": [
+          1.4348114728927612,
+          1.497140884399414,
+          1.434880018234253,
+          1.402381420135498,
+          2.8489983081817627,
+          2.8489983081817627
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "action": {
+      "left_arm": {
+        "min": [
+          -1.5419785976409912,
+          -0.19808706641197205,
+          -1.5707963705062866,
+          -2.3319737911224365,
+          -1.5707963705062866,
+          -0.6781747937202454,
+          -0.5912933349609375
+        ],
+        "max": [
+          0.339247465133667,
+          0.6958394646644592,
+          1.2758116722106934,
+          0.7663928866386414,
+          0.6010454297065735,
+          0.9854870438575745,
+          0.6981316804885864
+        ],
+        "mean": [
+          -0.27380800247192383,
+          0.27597638964653015,
+          -0.37349793314933777,
+          -1.6559933423995972,
+          -0.13736535608768463,
+          0.29294726252555847,
+          -0.005824246443808079
+        ],
+        "std": [
+          0.27345916628837585,
+          0.07336905598640442,
+          0.5032736659049988,
+          0.696763277053833,
+          0.4247623383998871,
+          0.24443820118904114,
+          0.23588497936725616
+        ],
+        "q01": [
+          -1.2448639106750488,
+          0.12081794798374176,
+          -1.4095649719238281,
+          -2.2054867362976074,
+          -1.5707963705062866,
+          -0.40510578513145445,
+          -0.34792898654937743
+        ],
+        "q99": [
+          0.08568856686353694,
+          0.4764096879959107,
+          1.1983782052993774,
+          0.6938967108726501,
+          0.3001057207584381,
+          0.9169320225715639,
+          0.5328076863288891
+        ]
+      },
+      "left_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.7413919568061829,
+          -0.1634250432252884,
+          -0.7413044571876526,
+          -0.687811553478241,
+          -1.5692328214645386,
+          -1.5692328214645386
+        ],
+        "std": [
+          1.299504280090332,
+          1.4938620328903198,
+          1.2995966672897339,
+          1.3312314748764038,
+          2.5481374263763428,
+          2.5481374263763428
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "left_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "neck": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "right_arm": {
+        "min": [
+          -0.7609559297561646,
+          -0.5897848010063171,
+          -0.13661164045333862,
+          -2.1815736293792725,
+          -0.6809128522872925,
+          -0.7053864002227783,
+          -0.6164834499359131
+        ],
+        "max": [
+          0.363760769367218,
+          0.464921772480011,
+          1.2909587621688843,
+          0.8227723836898804,
+          1.5061734914779663,
+          0.9927297830581665,
+          0.6635355353355408
+        ],
+        "mean": [
+          -0.20635083317756653,
+          -0.17740032076835632,
+          0.45279860496520996,
+          -1.6069952249526978,
+          0.0653856173157692,
+          -0.1357136219739914,
+          -0.009931313805282116
+        ],
+        "std": [
+          0.2129511535167694,
+          0.175509974360466,
+          0.26751890778541565,
+          0.45712384581565857,
+          0.5241751670837402,
+          0.2693447172641754,
+          0.20302622020244598
+        ],
+        "q01": [
+          -0.5830041313171387,
+          -0.44531570076942445,
+          0.04082439854741096,
+          -2.0516329193115235,
+          -0.5222506451606751,
+          -0.6403132677078247,
+          -0.3627131497859955
+        ],
+        "q99": [
+          0.2731454372406006,
+          0.4016782689094546,
+          1.2265138626098633,
+          0.6918689608573914,
+          1.3290673494338998,
+          0.8524364829063416,
+          0.45168524980545044
+        ]
+      },
+      "right_hand": {
+        "min": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "max": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ],
+        "mean": [
+          -0.40342941880226135,
+          0.1162116676568985,
+          -0.4033419191837311,
+          -0.43629777431488037,
+          -0.8933074474334717,
+          -0.8933074474334717
+        ],
+        "std": [
+          1.4348500967025757,
+          1.4971321821212769,
+          1.434924602508545,
+          1.4023455381393433,
+          2.849017381668091,
+          2.849017381668091
+        ],
+        "q01": [
+          -1.5,
+          -1.5,
+          -1.5,
+          -1.5,
+          -3.0,
+          -3.0
+        ],
+        "q99": [
+          1.5,
+          1.5,
+          1.5,
+          1.5,
+          3.0,
+          3.0
+        ]
+      },
+      "right_leg": {
+        "min": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0,
+          0.0
+        ]
+      },
+      "waist": {
+        "min": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "max": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "mean": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "std": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q01": [
+          0.0,
+          0.0,
+          0.0
+        ],
+        "q99": [
+          0.0,
+          0.0,
+          0.0
+        ]
+      }
+    },
+    "relative_action": {}
+  }
+}

gr00t/80_20_gr00t/checkpoint-9000/experiment_cfg/final_model_config.json

@@ -0,0 +1,53 @@
+{
+  "model_type": "Gr00tN1d6",
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Eagle-Block2A-2B-v2",
+  "backbone_model_type": "eagle",
+  "model_revision": null,
+  "tune_top_llm_layers": 4,
+  "backbone_embedding_dim": 2048,
+  "tune_llm": true,
+  "tune_visual": false,
+  "select_layer": 16,
+  "reproject_vision": false,
+  "use_flash_attention": true,
+  "load_bf16": true,
+  "collator_overwrite_image_inputs": false,
+  "eagle_collator": true,
+  "backbone_trainable_params_fp32": true,
+  "apply_sincos_state_encoding": true,
+  "use_relative_action": true,
+  "max_state_dim": 128,
+  "max_action_dim": 128,
+  "action_horizon": 50,
+  "hidden_size": 1024,
+  "input_embedding_dim": 1536,
+  "add_pos_embed": true,
+  "attn_dropout": 0.2,
+  "use_vlln": true,
+  "max_seq_len": 1024,
+  "use_alternate_vl_dit": true,
+  "attend_text_every_n_blocks": 2,
+  "diffusion_model_cfg": {
+    "attention_head_dim": 48,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "interleave_self_attention": true,
+    "norm_type": "ada_norm",
+    "num_attention_heads": 32,
+    "num_layers": 32,
+    "output_dim": 1024,
+    "positional_embeddings": null
+  },
+  "num_inference_timesteps": 4,
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_timestep_buckets": 1000,
+  "tune_projector": true,
+  "tune_diffusion_model": true,
+  "tune_vlln": true,
+  "state_dropout_prob": 0.0,
+  "state_additive_noise_scale": 0.0,
+  "max_num_embodiments": 32
+}

gr00t/80_20_gr00t/checkpoint-9000/model-00001-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c1ff220d1a27ecc465921748e45853d99df2c60d20bd01ac3fc8fd1b87162066
+size 4967164296

gr00t/80_20_gr00t/checkpoint-9000/model-00002-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e49bb5b3e3a8826fafdb7bcae72a5d87e13795b4798bc0149d23884d912b7a73
+size 4982025552

gr00t/80_20_gr00t/checkpoint-9000/model-00003-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90532fbb3c4f7c914452dd770c0fd572d549c0c0ae3aa3d139a5e70e9ad99b9e
+size 2314751176

gr00t/80_20_gr00t/checkpoint-9000/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67e34a85056607e2b5a115d95d3adf0ff7ab83c69ce3f488b6b8476ce3b241f3
+size 20277711680