dagger it1 checkpoint-750

checkpoint-750/config.json

@@ -0,0 +1,88 @@
+{
+  "action_horizon": 40,
+  "add_pos_embed": true,
+  "apply_sincos_state_encoding": false,
+  "architectures": [
+    "Gr00tN1d7"
+  ],
+  "attn_dropout": 0.2,
+  "attn_implementation": null,
+  "backbone_embedding_dim": 2048,
+  "backbone_trainable_params_fp32": true,
+  "color_jitter_params": {
+    "brightness": 0.3,
+    "contrast": 0.4,
+    "hue": 0.08,
+    "saturation": 0.5
+  },
+  "crop_fraction": 1.0,
+  "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
+  },
+  "dtype": "bfloat16",
+  "exclude_state": false,
+  "formalize_language": true,
+  "hidden_size": 1024,
+  "image_crop_size": [
+    230,
+    230
+  ],
+  "image_target_size": [
+    256,
+    256
+  ],
+  "letter_box_transform": false,
+  "load_bf16": false,
+  "max_action_dim": 132,
+  "max_num_embodiments": 32,
+  "max_seq_len": 1024,
+  "max_state_dim": 132,
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Cosmos-Reason2-2B",
+  "model_type": "Gr00tN1d7",
+  "noise_beta_alpha": 1.5,
+  "noise_beta_beta": 1.0,
+  "noise_s": 0.999,
+  "num_inference_timesteps": 4,
+  "num_timestep_buckets": 1000,
+  "random_history_crop": true,
+  "random_rotation_angle": 0,
+  "reproject_vision": false,
+  "rtc_ramp_rate": 6.0,
+  "select_layer": 16,
+  "shortest_image_edge": 256,
+  "state_dropout_prob": 0.2,
+  "state_gaussian_noise_std": 0.0,
+  "transformers_version": "4.57.3",
+  "tune_diffusion_model": true,
+  "tune_linear": true,
+  "tune_llm": false,
+  "tune_projector": true,
+  "tune_top_llm_layers": 0,
+  "tune_visual": false,
+  "tune_vlln": true,
+  "use_albumentations": true,
+  "use_alternate_vl_dit": true,
+  "use_flash_attention": true,
+  "use_future_tokens": false,
+  "use_mean_std": false,
+  "use_percentiles": false,
+  "use_vl_self_attention": true,
+  "use_vlln": true,
+  "vl_self_attention_cfg": {
+    "attention_head_dim": 64,
+    "dropout": 0.2,
+    "final_dropout": true,
+    "num_attention_heads": 32,
+    "num_layers": 4,
+    "positional_embeddings": null
+  }
+}

checkpoint-750/embodiment_id.json

@@ -0,0 +1,57 @@
+{
+  "robocasa_panda_omron": 13,
+  "oxe_droid": 17,
+  "oxe_fractal": 18,
+  "oxe_language_table": 19,
+  "oxe_bridge": 20,
+  "unknown": 22,
+  "gr1_unified": 20,
+  "agibot": 26,
+  "sim_behavior_r1_pro": 23,
+  "xdof": 24,
+  "xdof_oss_data": 25,
+  "unitree_g1_full_body_with_waist_height_nav_cmd": 25,
+  "real_r1_pro_sharpa": 27,
+  "real_r1_pro_sharpa_add_view": 27,
+  "real_r1_pro_sharpa_relative_arm_joint": 26,
+  "real_r1_pro_sharpa_delta_eef": 26,
+  "real_r1_pro_sharpa_absolute_eef": 26,
+  "real_r1_pro_sharpa_meanstd": 26,
+  "real_r1_pro_sharpa_relative_eef": 26,
+  "real_r1_pro_sharpa_relative_eef_add_view": 26,
+  "real_r1_pro_sharpa_relative_eef_relative_hand": 26,
+  "real_r1_pro_sharpa_relative_eef_human": 26,
+  "real_r1_pro_sharpa_relative_eef_human_add_view": 26,
+  "real_r1_pro_sharpa_relative_eef_human_relative_hand": 26,
+  "real_r1_pro_sharpa_relative_eef_egodex": 26,
+  "real_r1_pro_sharpa_relative_eef_egodex_relative_hand": 26,
+  "real_r1_pro_sharpa_relative_eef_egodex_wrist_only": 26,
+  "real_r1_pro_sharpa_relative_eef_maxinsights": 26,
+  "real_r1_pro_sharpa_relative_eef_maxinsights_relative_hand": 26,
+  "real_r1_pro_sharpa_relative_eef_mecka": 26,
+  "real_r1_pro_sharpa_relative_eef_mecka_relative_hand": 26,
+  "real_g1_relative_eef_absolute_joints": 25,
+  "real_g1_relative_eef_absolute_joints_wrist_cam": 25,
+  "real_g1_relative_eef_relative_joints": 25,
+  "real_r1_pro_sharpa_relative_eef_relative_hand_relative_joint": 26,
+  "real_r1_pro_sharpa_relative_joint": 29,
+  "oxe_droid_relative_eef_relative_joint": 24,
+  "oxe_droid_relative_eef_relative_joint_swapped": 24,
+  "oxe_droid_relative_eef_relative_joint_upweight_z": 24,
+  "oxe_droid_relative_eef_relative_joint_upweight_z_swapped": 24,
+  "oxe_droid_relative_eef_relative_joint_3view": 24,
+  "oxe_droid_relative_eef_relative_joint_3view_swapped": 24,
+  "oxe_droid_relative_eef": 24,
+  "oxe_droid_joint_position_relative": 24,
+  "xdof_relative_eef_relative_joint": 27,
+  "xdof_relative_eef_relative_joint_subtask": 27,
+  "xdof_relative_eef": 27,
+  "xdof_relative_joint": 28,
+  "simpler_env_google": 0,
+  "simpler_env_widowx": 1,
+  "libero_sim": 2,
+  "droid_sim": 3,
+  "unitree_g1_sonic": 11,
+  "new_embodiment": 10,
+  "robocasa_gr1_tabletop": 10
+}

checkpoint-750/experiment_cfg/conf.yaml

@@ -0,0 +1,230 @@
+load_config_path: null
+model:
+  model_type: Gr00tN1d7
+  model_dtype: bfloat16
+  model_name: nvidia/Cosmos-Reason2-2B
+  backbone_model_type: qwen
+  model_revision: null
+  tune_top_llm_layers: 0
+  backbone_embedding_dim: 2048
+  tune_llm: false
+  tune_visual: false
+  select_layer: 12
+  reproject_vision: false
+  use_flash_attention: true
+  load_bf16: false
+  backbone_trainable_params_fp32: true
+  image_crop_size:
+  - 230
+  - 230
+  image_target_size:
+  - 256
+  - 256
+  shortest_image_edge: null
+  crop_fraction: null
+  random_rotation_angle: null
+  color_jitter_params:
+    brightness: 0.2
+    contrast: 0.2
+    saturation: 0.2
+    hue: 0.03
+  use_albumentations_transforms: true
+  extra_augmentation_config: null
+  formalize_language: true
+  apply_sincos_state_encoding: false
+  use_percentiles: true
+  use_relative_action: true
+  max_state_dim: 132
+  max_action_dim: 132
+  action_horizon: 40
+  hidden_size: 1024
+  input_embedding_dim: 1536
+  state_history_length: 1
+  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: 16
+    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.2
+  exclude_state: false
+  use_mean_std: false
+  max_num_embodiments: 32
+data:
+  datasets:
+  - dataset_paths:
+    - /root/screwing_train/datasets/screwing35_follower_train
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.6
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  - dataset_paths:
+    - /root/screwing_train/datasets/dagger_human
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.3
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  - dataset_paths:
+    - /root/screwing_train/datasets/dagger_auto
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.1
+    dataset_type: physical_embodiment
+    val_dataset_path: null
+  modality_configs:
+    new_embodiment:
+      video:
+        delta_indices:
+        - 0
+        modality_keys:
+        - cam_left_head
+        - cam_left_wrist
+        - cam_right_wrist
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      state:
+        delta_indices:
+        - 0
+        modality_keys:
+        - arm_left
+        - arm_right
+        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:
+        - arm_left
+        - arm_right
+        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
+      language:
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.human.primitive_instruction
+        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: true
+  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: /root/screwing_train/runs/groot_dagger_it1
+  experiment_name: null
+  max_steps: 6000
+  global_batch_size: 128
+  batch_size: null
+  gradient_accumulation_steps: 1
+  learning_rate: 3.0e-05
+  lr_scheduler_type: cosine
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.03
+  warmup_steps: 0
+  max_grad_norm: 1.0
+  optim: adamw_torch
+  start_from_checkpoint: /root/screwing_train/runs/16d_30k/checkpoint-30000
+  skip_weight_loading: false
+  tf32: true
+  fp16: false
+  bf16: true
+  eval_bf16: true
+  logging_steps: 10
+  save_steps: 250
+  save_total_limit: 24
+  save_vl_model: false
+  save_only_model: true
+  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: 2
+  dataloader_num_workers: 32
+  remove_unused_columns: false
+  use_wandb: true
+  wandb_project: screwing_hil
+  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: 6000
+save_steps: 250

checkpoint-750/experiment_cfg/config.yaml

@@ -0,0 +1,257 @@
+!!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:
+    - /root/screwing_train/datasets/screwing35_follower_train
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.6
+    val_dataset_path: null
+  - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
+    dataset_paths:
+    - /root/screwing_train/datasets/dagger_human
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.3
+    val_dataset_path: null
+  - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
+    dataset_paths:
+    - /root/screwing_train/datasets/dagger_auto
+    dataset_type: physical_embodiment
+    embodiment_tag: new_embodiment
+    mix_ratio: 0.1
+    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
+        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:
+        - arm_left
+        - arm_right
+        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.primitive_instruction
+        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:
+        - arm_left
+        - arm_right
+        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:
+        - cam_left_head
+        - cam_left_wrist
+        - cam_right_wrist
+        sin_cos_embedding_keys: null
+  mode: single_turn
+  multiprocessing_context: fork
+  num_shards_per_epoch: 100000
+  override_pretraining_statistics: true
+  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_n1d7.Gr00tN1d7Config
+  _attn_implementation_internal: null
+  _commit_hash: null
+  _name_or_path: ''
+  _output_attentions: false
+  add_cross_attention: false
+  architectures: null
+  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.2
+    contrast: 0.2
+    hue: 0.03
+    saturation: 0.2
+  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: 16
+    output_dim: 1024
+    positional_embeddings: null
+  diversity_penalty: 0.0
+  do_sample: false
+  dtype: null
+  early_stopping: false
+  encoder_no_repeat_ngram_size: 0
+  eos_token_id: null
+  exponential_decay_length_penalty: null
+  extra_augmentation_config: 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/Cosmos-Reason2-2B
+  no_repeat_ngram_size: 0
+  num_beam_groups: 1
+  num_beams: 1
+  num_return_sequences: 1
+  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.2
+  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
+  torchscript: false
+  transformers_version: null
+  tune_diffusion_model: true
+  tune_llm: false
+  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: 32
+  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: 128
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: false
+  learning_rate: 3.0e-05
+  logging_steps: 10
+  lr_scheduler_type: cosine
+  max_concurrent_uploads: 2
+  max_grad_norm: 1.0
+  max_retries: 3
+  max_steps: 6000
+  num_gpus: 2
+  open_loop_eval_plot_indices: null
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_traj_ids:
+  - 0
+  optim: adamw_torch
+  output_dir: /root/screwing_train/runs/groot_dagger_it1
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_only_model: true
+  save_steps: 250
+  save_total_limit: 24
+  save_vl_model: false
+  skip_weight_loading: false
+  start_from_checkpoint: /root/screwing_train/runs/16d_30k/checkpoint-30000
+  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: screwing_hil
+  warmup_ratio: 0.03
+  warmup_steps: 0
+  weight_decay: 1.0e-05

checkpoint-750/experiment_cfg/dataset_statistics.json

@@ -0,0 +1,257 @@
+{
+  "new_embodiment": {
+    "state": {
+      "arm_left": {
+        "min": [
+          -0.3512935936450958,
+          -0.0003355582884978503,
+          -0.4982561469078064,
+          -2.0985336303710938,
+          -0.30987611413002014,
+          -0.390062540769577,
+          -0.756526529788971,
+          -0.0015339808305725455
+        ],
+        "max": [
+          0.4587441384792328,
+          0.4674566686153412,
+          0.2403915673494339,
+          -0.8097261190414429,
+          0.3366488516330719,
+          0.7219297289848328,
+          0.09214304387569427,
+          1.0001554489135742
+        ],
+        "mean": [
+          0.04355059830193595,
+          0.10864019471928726,
+          -0.10750242693181013,
+          -1.7219742652123884,
+          0.013241997072442023,
+          0.24038831582343306,
+          -0.3289317812120216,
+          0.21972585904618444
+        ],
+        "std": [
+          0.201977025178683,
+          0.08930106217863336,
+          0.1703593429065693,
+          0.2290651088739729,
+          0.13351265238651933,
+          0.2640519575529048,
+          0.2311836246595438,
+          0.3695154517778703
+        ],
+        "q01": [
+          -0.28752409607172014,
+          0.0,
+          -0.4731247413158417,
+          -2.013697385787964,
+          -0.2523278594017029,
+          -0.2879210114479065,
+          -0.7209579944610596,
+          0.0
+        ],
+        "q99": [
+          0.4115143120288849,
+          0.4564431607723236,
+          0.18591128289699554,
+          -0.932169679403305,
+          0.2769750881195063,
+          0.7118869423866272,
+          0.06595534831285477,
+          0.9970875382423401
+        ]
+      },
+      "arm_right": {
+        "min": [
+          -0.9125747680664062,
+          -0.46813976764678955,
+          -0.08915065228939056,
+          -2.0727076530456543,
+          -0.9404261112213135,
+          -0.15235544741153717,
+          -0.053947482258081436,
+          0.0
+        ],
+        "max": [
+          0.4171828627586365,
+          -0.031806133687496185,
+          1.0656253099441528,
+          -1.032512903213501,
+          0.13881327211856842,
+          1.1778095960617065,
+          0.8715579509735107,
+          1.277806043624878
+        ],
+        "mean": [
+          -0.29372113303835257,
+          -0.25370953344792585,
+          0.3190493214782265,
+          -1.6728425137445484,
+          -0.24982502985306895,
+          0.49206255521233844,
+          0.42576361554110886,
+          0.4498966951422322
+        ],
+        "std": [
+          0.24819130455260907,
+          0.05331376467400995,
+          0.17461899684902152,
+          0.1924666931220957,
+          0.18633659471924663,
+          0.28551332367395904,
+          0.17700279003797392,
+          0.3465478824834864
+        ],
+        "q01": [
+          -0.8353484272956848,
+          -0.43082091212272644,
+          -0.01853959634900093,
+          -2.0143797326087953,
+          -0.7463775277137756,
+          -0.06663229316473007,
+          0.07988695055246353,
+          0.0015339808305725455
+        ],
+        "q99": [
+          0.28748956322669983,
+          -0.14874820411205292,
+          0.8437014222145081,
+          -1.2486603260040283,
+          0.12267052382230759,
+          1.0415489673614502,
+          0.8113635182380676,
+          0.8682331442832947
+        ]
+      }
+    },
+    "action": {
+      "arm_left": {
+        "min": [
+          -0.3512935936450958,
+          -0.0003355582884978503,
+          -0.49900805950164795,
+          -2.0985336303710938,
+          -0.30987611413002014,
+          -0.390062540769577,
+          -0.756526529788971,
+          -0.0015339808305725455
+        ],
+        "max": [
+          0.4587441384792328,
+          0.4674566686153412,
+          0.2403915673494339,
+          -0.8097261190414429,
+          0.3393723964691162,
+          0.7219297289848328,
+          0.09214304387569427,
+          1.2387962341308594
+        ],
+        "mean": [
+          0.04354117550648276,
+          0.10888600258373515,
+          -0.10755284350603393,
+          -1.7220248298756244,
+          0.013494755258123084,
+          0.24065059866611607,
+          -0.32936498765478983,
+          0.22581150584984677
+        ],
+        "std": [
+          0.20216011477897333,
+          0.08923828878575918,
+          0.170728239388772,
+          0.22946179527618987,
+          0.13395482743250087,
+          0.2638166849754727,
+          0.23123107266252962,
+          0.3869448499567072
+        ],
+        "q01": [
+          -0.2880094051361084,
+          0.0,
+          -0.4840488076210022,
+          -2.013697385787964,
+          -0.2523278594017029,
+          -0.2877771854400635,
+          -0.7209579944610596,
+          0.0
+        ],
+        "q99": [
+          0.41167011857032776,
+          0.4564431607723236,
+          0.1860191375017166,
+          -0.9313069581985474,
+          0.2917774844169608,
+          0.7118869423866272,
+          0.06595534831285477,
+          1.23726224899292
+        ]
+      },
+      "arm_right": {
+        "min": [
+          -0.9125747680664062,
+          -0.46813976764678955,
+          -0.08915065228939056,
+          -2.0727076530456543,
+          -0.9404261112213135,
+          -0.15235544741153717,
+          -0.053947482258081436,
+          0.0
+        ],
+        "max": [
+          0.4171828627586365,
+          -0.031806133687496185,
+          1.0656253099441528,
+          -1.032512903213501,
+          0.13881327211856842,
+          1.1778095960617065,
+          0.8715579509735107,
+          1.3185632228851318
+        ],
+        "mean": [
+          -0.2942656780309203,
+          -0.25352573533534634,
+          0.31921004536062647,
+          -1.6735023107714255,
+          -0.2505161447075832,
+          0.4924302416916429,
+          0.42694812914012836,
+          0.5413754129270169
+        ],
+        "std": [
+          0.2486902262630544,
+          0.05344843369680529,
+          0.17478912708816832,
+          0.19249047992758175,
+          0.18634200598780004,
+          0.28484720108338507,
+          0.17626246445564833,
+          0.47102636928109104
+        ],
+        "q01": [
+          -0.8353484272956848,
+          -0.43082091212272644,
+          -0.01853959634900093,
+          -2.014399766921997,
+          -0.7463775277137756,
+          -0.06663229316473007,
+          0.07988695055246353,
+          0.0015339808305725455
+        ],
+        "q99": [
+          0.288448303937912,
+          -0.14868827164173126,
+          0.8437014222145081,
+          -1.2486603260040283,
+          0.12267052382230759,
+          1.0415489673614502,
+          0.8113635182380676,
+          1.3185632228851318
+        ]
+      }
+    },
+    "relative_action": {}
+  }
+}

checkpoint-750/experiment_cfg/final_model_config.json

@@ -0,0 +1,55 @@
+{
+  "model_type": "Gr00tN1d7",
+  "model_dtype": "bfloat16",
+  "model_name": "nvidia/Cosmos-Reason2-2B",
+  "backbone_model_type": "qwen",
+  "model_revision": null,
+  "tune_top_llm_layers": 0,
+  "backbone_embedding_dim": 2048,
+  "tune_llm": false,
+  "tune_visual": false,
+  "select_layer": 16,
+  "reproject_vision": false,
+  "use_flash_attention": true,
+  "load_bf16": false,
+  "backbone_trainable_params_fp32": true,
+  "extra_augmentation_config": null,
+  "apply_sincos_state_encoding": false,
+  "use_percentiles": false,
+  "use_relative_action": false,
+  "max_state_dim": 132,
+  "max_action_dim": 132,
+  "action_horizon": 40,
+  "hidden_size": 1024,
+  "input_embedding_dim": 1536,
+  "state_history_length": 1,
+  "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.2,
+  "exclude_state": false,
+  "use_mean_std": false,
+  "max_num_embodiments": 32
+}

checkpoint-750/model-00001-of-00002.safetensors

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

checkpoint-750/model-00002-of-00002.safetensors

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

checkpoint-750/processor_config.json

@@ -0,0 +1,1124 @@
+{
+  "processor_class": "Gr00tN1d7Processor",
+  "processor_kwargs": {
+    "modality_configs": {
+      "real_g1_relative_eef_relative_joints": {
+        "video": {
+          "delta_indices": [
+            -20,
+            0
+          ],
+          "modality_keys": [
+            "ego_view"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef_9d",
+            "right_wrist_eef_9d",
+            "left_hand",
+            "right_hand",
+            "left_arm",
+            "right_arm",
+            "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,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef_9d",
+            "right_wrist_eef_9d",
+            "left_hand",
+            "right_hand",
+            "left_arm",
+            "right_arm",
+            "waist",
+            "base_height_command",
+            "navigate_command"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef_9d"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef_9d"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_hand"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_hand"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_arm"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_arm"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "waist"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "base_height_command"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "navigate_command"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.task_description"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "real_r1_pro_sharpa_relative_eef_mecka": {
+        "video": {
+          "delta_indices": [
+            -30,
+            0
+          ],
+          "modality_keys": [
+            "ego_view_cropratio_res320x240_freq30"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_hand_joints"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_hand_joints"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "real_r1_pro_sharpa_relative_eef_human": {
+        "video": {
+          "delta_indices": [
+            -20,
+            0
+          ],
+          "modality_keys": [
+            "ego_view_res320x240_freq20",
+            "left_wrist_view_res320x240_freq20",
+            "right_wrist_view_res320x240_freq20"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_hand_joints"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_hand_joints"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "real_r1_pro_sharpa_relative_eef": {
+        "video": {
+          "delta_indices": [
+            -20,
+            0
+          ],
+          "modality_keys": [
+            "ego_view_res320x240_freq20",
+            "left_wrist_view_res320x240_freq20",
+            "right_wrist_view_res320x240_freq20"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_hand_joints"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_hand_joints"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "xdof_relative_eef_relative_joint": {
+        "video": {
+          "delta_indices": [
+            -30,
+            0
+          ],
+          "modality_keys": [
+            "top_camera-images-rgb_320_240",
+            "left_camera-images-rgb_320_240",
+            "right_camera-images-rgb_320_240"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_gripper_pos",
+            "right_gripper_pos",
+            "left_joint_pos",
+            "right_joint_pos"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_gripper_pos",
+            "right_gripper_pos",
+            "left_joint_pos",
+            "right_joint_pos"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_gripper_pos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_gripper_pos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_joint_pos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_joint_pos"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.task"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "real_r1_pro_sharpa_relative_eef_maxinsights": {
+        "video": {
+          "delta_indices": [
+            -30,
+            0
+          ],
+          "modality_keys": [
+            "ego_view_cropratio_res320x240_freq30"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_hand_joints",
+            "right_hand_joints"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_hand_joints"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_hand_joints"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.coarse_action"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "xdof_relative_eef_relative_joint_subtask": {
+        "video": {
+          "delta_indices": [
+            -30,
+            0
+          ],
+          "modality_keys": [
+            "top_camera-images-rgb_320_240",
+            "left_camera-images-rgb_320_240",
+            "right_camera-images-rgb_320_240"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_gripper_pos",
+            "right_gripper_pos",
+            "left_joint_pos",
+            "right_joint_pos"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "left_wrist_eef",
+            "right_wrist_eef",
+            "left_gripper_pos",
+            "right_gripper_pos",
+            "left_joint_pos",
+            "right_joint_pos"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "left_wrist_eef"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "right_wrist_eef"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_gripper_pos"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_gripper_pos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "left_joint_pos"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "right_joint_pos"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.sub_task"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "oxe_droid_relative_eef_relative_joint": {
+        "video": {
+          "delta_indices": [
+            -15,
+            0
+          ],
+          "modality_keys": [
+            "exterior_image_1_left",
+            "wrist_image_left"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "eef_9d",
+            "gripper_position",
+            "joint_position"
+          ],
+          "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,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39
+          ],
+          "modality_keys": [
+            "eef_9d",
+            "gripper_position",
+            "joint_position"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "rep": "RELATIVE",
+              "type": "EEF",
+              "format": "XYZ_ROT6D",
+              "state_key": "eef_9d"
+            },
+            {
+              "rep": "ABSOLUTE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "gripper_position"
+            },
+            {
+              "rep": "RELATIVE",
+              "type": "NON_EEF",
+              "format": "DEFAULT",
+              "state_key": "joint_position"
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.language.language_instruction"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      },
+      "new_embodiment": {
+        "video": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "cam_left_head",
+            "cam_left_wrist",
+            "cam_right_wrist"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "arm_left",
+            "arm_right"
+          ],
+          "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": [
+            "arm_left",
+            "arm_right"
+          ],
+          "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
+            }
+          ]
+        },
+        "language": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "annotation.human.primitive_instruction"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        }
+      }
+    },
+    "image_crop_size": [
+      230,
+      230
+    ],
+    "image_target_size": [
+      256,
+      256
+    ],
+    "use_albumentations": true,
+    "random_rotation_angle": 0,
+    "color_jitter_params": {
+      "brightness": 0.2,
+      "contrast": 0.2,
+      "saturation": 0.2,
+      "hue": 0.03
+    },
+    "shortest_image_edge": 256,
+    "crop_fraction": 1.0,
+    "letter_box_transform": false,
+    "model_name": "nvidia/Cosmos-Reason2-2B",
+    "model_type": "qwen",
+    "formalize_language": true,
+    "max_state_dim": 132,
+    "max_action_dim": 132,
+    "max_action_horizon": 40,
+    "use_percentiles": false,
+    "use_mean_std": false,
+    "clip_outliers": true,
+    "apply_sincos_state_encoding": false,
+    "use_relative_action": true,
+    "exclude_state": false,
+    "state_dropout_prob": 0.2
+  }
+}

checkpoint-750/trainer_state.json

@@ -0,0 +1,484 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.125,
+  "eval_steps": 500,
+  "global_step": 750,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "grad_norm": 1.1998956203460693,
+      "learning_rate": 1.5e-06,
+      "loss": 0.0948,
+      "step": 10
+    },
+    {
+      "grad_norm": 1.149483561515808,
+      "learning_rate": 3.1666666666666667e-06,
+      "loss": 0.1133,
+      "step": 20
+    },
+    {
+      "grad_norm": 0.5785359144210815,
+      "learning_rate": 4.833333333333333e-06,
+      "loss": 0.0921,
+      "step": 30
+    },
+    {
+      "grad_norm": 0.5978452563285828,
+      "learning_rate": 6.5000000000000004e-06,
+      "loss": 0.0424,
+      "step": 40
+    },
+    {
+      "grad_norm": 0.5148229598999023,
+      "learning_rate": 8.166666666666666e-06,
+      "loss": 0.0314,
+      "step": 50
+    },
+    {
+      "grad_norm": 0.456398606300354,
+      "learning_rate": 9.833333333333333e-06,
+      "loss": 0.0311,
+      "step": 60
+    },
+    {
+      "grad_norm": 0.2702748775482178,
+      "learning_rate": 1.1500000000000002e-05,
+      "loss": 0.0249,
+      "step": 70
+    },
+    {
+      "grad_norm": 0.2341022789478302,
+      "learning_rate": 1.3166666666666667e-05,
+      "loss": 0.0208,
+      "step": 80
+    },
+    {
+      "grad_norm": 0.24386702477931976,
+      "learning_rate": 1.4833333333333334e-05,
+      "loss": 0.0168,
+      "step": 90
+    },
+    {
+      "grad_norm": 0.15167136490345,
+      "learning_rate": 1.65e-05,
+      "loss": 0.0114,
+      "step": 100
+    },
+    {
+      "grad_norm": 0.12867538630962372,
+      "learning_rate": 1.8166666666666667e-05,
+      "loss": 0.015,
+      "step": 110
+    },
+    {
+      "grad_norm": 0.11734911054372787,
+      "learning_rate": 1.9833333333333332e-05,
+      "loss": 0.0108,
+      "step": 120
+    },
+    {
+      "grad_norm": 0.10739504545927048,
+      "learning_rate": 2.15e-05,
+      "loss": 0.0085,
+      "step": 130
+    },
+    {
+      "grad_norm": 0.11796178668737411,
+      "learning_rate": 2.316666666666667e-05,
+      "loss": 0.0108,
+      "step": 140
+    },
+    {
+      "grad_norm": 0.1394447386264801,
+      "learning_rate": 2.483333333333333e-05,
+      "loss": 0.0111,
+      "step": 150
+    },
+    {
+      "grad_norm": 0.1239742562174797,
+      "learning_rate": 2.65e-05,
+      "loss": 0.01,
+      "step": 160
+    },
+    {
+      "grad_norm": 0.15108101069927216,
+      "learning_rate": 2.8166666666666666e-05,
+      "loss": 0.0088,
+      "step": 170
+    },
+    {
+      "grad_norm": 0.10625599324703217,
+      "learning_rate": 2.9833333333333335e-05,
+      "loss": 0.0073,
+      "step": 180
+    },
+    {
+      "grad_norm": 0.18963077664375305,
+      "learning_rate": 2.9999822989428527e-05,
+      "loss": 0.0086,
+      "step": 190
+    },
+    {
+      "grad_norm": 0.13748835027217865,
+      "learning_rate": 2.9999211106396838e-05,
+      "loss": 0.008,
+      "step": 200
+    },
+    {
+      "grad_norm": 0.11299511790275574,
+      "learning_rate": 2.9998162183413746e-05,
+      "loss": 0.0084,
+      "step": 210
+    },
+    {
+      "grad_norm": 0.09556759148836136,
+      "learning_rate": 2.9996676251042288e-05,
+      "loss": 0.0075,
+      "step": 220
+    },
+    {
+      "grad_norm": 0.13729563355445862,
+      "learning_rate": 2.9994753352578854e-05,
+      "loss": 0.0078,
+      "step": 230
+    },
+    {
+      "grad_norm": 0.13592101633548737,
+      "learning_rate": 2.999239354405195e-05,
+      "loss": 0.0087,
+      "step": 240
+    },
+    {
+      "grad_norm": 0.13999280333518982,
+      "learning_rate": 2.9989596894220574e-05,
+      "loss": 0.0072,
+      "step": 250
+    },
+    {
+      "grad_norm": 0.20841121673583984,
+      "learning_rate": 2.998636348457219e-05,
+      "loss": 0.0078,
+      "step": 260
+    },
+    {
+      "grad_norm": 0.1364365965127945,
+      "learning_rate": 2.998269340932036e-05,
+      "loss": 0.0086,
+      "step": 270
+    },
+    {
+      "grad_norm": 0.1565951108932495,
+      "learning_rate": 2.9978586775402017e-05,
+      "loss": 0.0066,
+      "step": 280
+    },
+    {
+      "grad_norm": 0.1358197182416916,
+      "learning_rate": 2.9974043702474322e-05,
+      "loss": 0.0057,
+      "step": 290
+    },
+    {
+      "grad_norm": 0.11862654238939285,
+      "learning_rate": 2.9969064322911198e-05,
+      "loss": 0.0074,
+      "step": 300
+    },
+    {
+      "grad_norm": 0.13869383931159973,
+      "learning_rate": 2.9963648781799457e-05,
+      "loss": 0.0073,
+      "step": 310
+    },
+    {
+      "grad_norm": 0.10812313109636307,
+      "learning_rate": 2.9957797236934595e-05,
+      "loss": 0.0055,
+      "step": 320
+    },
+    {
+      "grad_norm": 0.10657292604446411,
+      "learning_rate": 2.995150985881616e-05,
+      "loss": 0.0066,
+      "step": 330
+    },
+    {
+      "grad_norm": 0.1428467333316803,
+      "learning_rate": 2.9944786830642822e-05,
+      "loss": 0.0075,
+      "step": 340
+    },
+    {
+      "grad_norm": 0.2037973701953888,
+      "learning_rate": 2.9937628348307015e-05,
+      "loss": 0.0068,
+      "step": 350
+    },
+    {
+      "grad_norm": 0.17431488633155823,
+      "learning_rate": 2.9930034620389217e-05,
+      "loss": 0.0085,
+      "step": 360
+    },
+    {
+      "grad_norm": 0.11183494329452515,
+      "learning_rate": 2.9922005868151907e-05,
+      "loss": 0.0075,
+      "step": 370
+    },
+    {
+      "grad_norm": 0.15082184970378876,
+      "learning_rate": 2.991354232553308e-05,
+      "loss": 0.0062,
+      "step": 380
+    },
+    {
+      "grad_norm": 0.18662261962890625,
+      "learning_rate": 2.9904644239139453e-05,
+      "loss": 0.0065,
+      "step": 390
+    },
+    {
+      "grad_norm": 0.08828644454479218,
+      "learning_rate": 2.9895311868239293e-05,
+      "loss": 0.0063,
+      "step": 400
+    },
+    {
+      "grad_norm": 0.09585582464933395,
+      "learning_rate": 2.988554548475481e-05,
+      "loss": 0.0085,
+      "step": 410
+    },
+    {
+      "grad_norm": 0.1154177337884903,
+      "learning_rate": 2.9875345373254302e-05,
+      "loss": 0.0062,
+      "step": 420
+    },
+    {
+      "grad_norm": 0.13848039507865906,
+      "learning_rate": 2.986471183094382e-05,
+      "loss": 0.0072,
+      "step": 430
+    },
+    {
+      "grad_norm": 0.11135748773813248,
+      "learning_rate": 2.9853645167658494e-05,
+      "loss": 0.0052,
+      "step": 440
+    },
+    {
+      "grad_norm": 0.15800616145133972,
+      "learning_rate": 2.9842145705853565e-05,
+      "loss": 0.0057,
+      "step": 450
+    },
+    {
+      "grad_norm": 0.11519774794578552,
+      "learning_rate": 2.9830213780594925e-05,
+      "loss": 0.0041,
+      "step": 460
+    },
+    {
+      "grad_norm": 0.06121053546667099,
+      "learning_rate": 2.9817849739549408e-05,
+      "loss": 0.0045,
+      "step": 470
+    },
+    {
+      "grad_norm": 0.11012984067201614,
+      "learning_rate": 2.980505394297461e-05,
+      "loss": 0.0047,
+      "step": 480
+    },
+    {
+      "grad_norm": 0.12504155933856964,
+      "learning_rate": 2.9791826763708432e-05,
+      "loss": 0.0043,
+      "step": 490
+    },
+    {
+      "grad_norm": 0.10626716166734695,
+      "learning_rate": 2.9778168587158196e-05,
+      "loss": 0.005,
+      "step": 500
+    },
+    {
+      "grad_norm": 0.100404754281044,
+      "learning_rate": 2.976407981128942e-05,
+      "loss": 0.0037,
+      "step": 510
+    },
+    {
+      "grad_norm": 0.11580510437488556,
+      "learning_rate": 2.9749560846614216e-05,
+      "loss": 0.0042,
+      "step": 520
+    },
+    {
+      "grad_norm": 0.1404998004436493,
+      "learning_rate": 2.973461211617935e-05,
+      "loss": 0.0055,
+      "step": 530
+    },
+    {
+      "grad_norm": 0.1066383644938469,
+      "learning_rate": 2.9719234055553884e-05,
+      "loss": 0.0055,
+      "step": 540
+    },
+    {
+      "grad_norm": 0.16468606889247894,
+      "learning_rate": 2.970342711281651e-05,
+      "loss": 0.0038,
+      "step": 550
+    },
+    {
+      "grad_norm": 0.1270781308412552,
+      "learning_rate": 2.9687191748542485e-05,
+      "loss": 0.0056,
+      "step": 560
+    },
+    {
+      "grad_norm": 0.13871441781520844,
+      "learning_rate": 2.967052843579021e-05,
+      "loss": 0.0079,
+      "step": 570
+    },
+    {
+      "grad_norm": 0.13760028779506683,
+      "learning_rate": 2.965343766008745e-05,
+      "loss": 0.0047,
+      "step": 580
+    },
+    {
+      "grad_norm": 0.23738646507263184,
+      "learning_rate": 2.9635919919417175e-05,
+      "loss": 0.0058,
+      "step": 590
+    },
+    {
+      "grad_norm": 0.12334813177585602,
+      "learning_rate": 2.9617975724203065e-05,
+      "loss": 0.006,
+      "step": 600
+    },
+    {
+      "grad_norm": 0.11787013709545135,
+      "learning_rate": 2.9599605597294635e-05,
+      "loss": 0.0042,
+      "step": 610
+    },
+    {
+      "grad_norm": 0.10075175762176514,
+      "learning_rate": 2.958081007395199e-05,
+      "loss": 0.0051,
+      "step": 620
+    },
+    {
+      "grad_norm": 0.12161751091480255,
+      "learning_rate": 2.956158970183024e-05,
+      "loss": 0.006,
+      "step": 630
+    },
+    {
+      "grad_norm": 0.1339486688375473,
+      "learning_rate": 2.954194504096355e-05,
+      "loss": 0.0053,
+      "step": 640
+    },
+    {
+      "grad_norm": 0.09474559128284454,
+      "learning_rate": 2.9521876663748787e-05,
+      "loss": 0.0069,
+      "step": 650
+    },
+    {
+      "grad_norm": 0.10630160570144653,
+      "learning_rate": 2.950138515492888e-05,
+      "loss": 0.0053,
+      "step": 660
+    },
+    {
+      "grad_norm": 0.1059340387582779,
+      "learning_rate": 2.9480471111575768e-05,
+      "loss": 0.0047,
+      "step": 670
+    },
+    {
+      "grad_norm": 0.13061285018920898,
+      "learning_rate": 2.9459135143073e-05,
+      "loss": 0.0044,
+      "step": 680
+    },
+    {
+      "grad_norm": 0.11301890015602112,
+      "learning_rate": 2.9437377871097974e-05,
+      "loss": 0.004,
+      "step": 690
+    },
+    {
+      "grad_norm": 0.11071354895830154,
+      "learning_rate": 2.9415199929603843e-05,
+      "loss": 0.0048,
+      "step": 700
+    },
+    {
+      "grad_norm": 0.10549204051494598,
+      "learning_rate": 2.939260196480103e-05,
+      "loss": 0.0053,
+      "step": 710
+    },
+    {
+      "grad_norm": 0.13164091110229492,
+      "learning_rate": 2.9369584635138386e-05,
+      "loss": 0.0051,
+      "step": 720
+    },
+    {
+      "grad_norm": 0.1299622356891632,
+      "learning_rate": 2.9346148611284025e-05,
+      "loss": 0.0057,
+      "step": 730
+    },
+    {
+      "grad_norm": 0.12443673610687256,
+      "learning_rate": 2.9322294576105785e-05,
+      "loss": 0.0054,
+      "step": 740
+    },
+    {
+      "grad_norm": 0.2119351178407669,
+      "learning_rate": 2.92980232246513e-05,
+      "loss": 0.006,
+      "step": 750
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 6000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 9223372036854775807,
+  "save_steps": 250,
+  "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": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-750/training_args.bin

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

checkpoint-750/wandb_config.json

@@ -0,0 +1 @@
+{"project": "screwing_hil", "run_id": "groot_dagger_it1"}