Upload folder using huggingface_hub

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": false,
+  "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
+}

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
+}

checkpoint-10000/experiment_cfg/conf.yaml

@@ -0,0 +1,211 @@
+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: false
+  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
+  extra_augmentation_config: null
+  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:
+    - ./demo_data/ps2/viis0002/cluttered_picking_v2
+    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:
+        - front
+        - wrist_color
+        - wrist_depth
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs: null
+      state:
+        delta_indices:
+        - 0
+        modality_keys:
+        - single_arm
+        - gripper
+        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:
+        - single_arm
+        - gripper
+        sin_cos_embedding_keys: null
+        mean_std_embedding_keys: null
+        action_configs:
+        - 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:
+        - annotation.human.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: ./output
+  experiment_name: null
+  max_steps: 10000
+  global_batch_size: 8
+  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: 5
+  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: 2
+  remove_unused_columns: false
+  use_wandb: false
+  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

checkpoint-10000/experiment_cfg/config.yaml

@@ -0,0 +1,245 @@
+!!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:
+    - ./demo_data/ps2/viis0002/cluttered_picking_v2
+    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: !!python/object/apply:gr00t.data.types.ActionRepresentation
+          - relative
+          state_key: null
+          type: &id002 !!python/object/apply:gr00t.data.types.ActionType
+          - non_eef
+        - !!python/object:gr00t.data.types.ActionConfig
+          format: *id001
+          rep: !!python/object/apply:gr00t.data.types.ActionRepresentation
+          - absolute
+          state_key: null
+          type: *id002
+        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:
+        - single_arm
+        - gripper
+        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.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:
+        - single_arm
+        - gripper
+        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:
+        - front
+        - wrist_color
+        - wrist_depth
+        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
+  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/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: 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: 2
+  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: 8
+  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: ./output
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_steps: 3000
+  save_total_limit: 5
+  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: false
+  wandb_project: finetune-gr00t-n1d6
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  weight_decay: 1.0e-05

checkpoint-10000/experiment_cfg/dataset_statistics.json

@@ -0,0 +1,824 @@
+{
+  "new_embodiment": {
+    "state": {
+      "single_arm": {
+        "min": [
+          -43.319149017333984,
+          -99.40652465820312,
+          -39.10285568237305,
+          48.36065673828125,
+          -34.70085525512695
+        ],
+        "max": [
+          11.829787254333496,
+          40.398475646972656,
+          100.0,
+          100.0,
+          56.43467712402344
+        ],
+        "mean": [
+          -4.440749271172309,
+          -13.308821339499419,
+          15.1325215701596,
+          94.23170722591506,
+          9.425153629722358
+        ],
+        "std": [
+          12.14406229008906,
+          36.99829120476212,
+          34.39080809147234,
+          9.257896912650017,
+          16.60771267620677
+        ],
+        "q01": [
+          -32.765687240006784,
+          -98.18120676093605,
+          -20.29557584031693,
+          67.62860396622403,
+          -5.580963820366954
+        ],
+        "q99": [
+          8.45665083715747,
+          31.670184474024747,
+          95.99070525799908,
+          99.69872470853615,
+          40.65240490687968
+        ]
+      },
+      "gripper": {
+        "min": [
+          0.5750798583030701
+        ],
+        "max": [
+          73.9297103881836
+        ],
+        "mean": [
+          9.998555719102907
+        ],
+        "std": [
+          12.050090684502234
+        ],
+        "q01": [
+          1.197928574688387
+        ],
+        "q99": [
+          47.81798753608932
+        ]
+      }
+    },
+    "action": {
+      "single_arm": {
+        "min": [
+          -43.884891510009766,
+          -99.5762710571289,
+          -43.259193420410156,
+          46.74245834350586,
+          -34.94505310058594
+        ],
+        "max": [
+          12.53313159942627,
+          41.949153900146484,
+          99.63685607910156,
+          100.0,
+          56.82539749145508
+        ],
+        "mean": [
+          -4.336263196603688,
+          -15.522390066565956,
+          13.258741117242543,
+          94.06182767417421,
+          9.397664254783678
+        ],
+        "std": [
+          12.372809314706336,
+          36.56167326493358,
+          34.92591755064374,
+          9.370667724934858,
+          16.699527746233734
+        ],
+        "q01": [
+          -33.256299413939324,
+          -98.9770363306071,
+          -23.248310714347284,
+          66.50771469329445,
+          -5.869576220557871
+        ],
+        "q99": [
+          8.985025713499576,
+          31.601930304179678,
+          93.41583007241056,
+          99.88448441811833,
+          40.89866939535895
+        ]
+      },
+      "gripper": {
+        "min": [
+          0.0
+        ],
+        "max": [
+          74.22003173828125
+        ],
+        "mean": [
+          7.439182423450768
+        ],
+        "std": [
+          12.415960562905878
+        ],
+        "q01": [
+          0.7503314920264533
+        ],
+        "q99": [
+          48.41172194516817
+        ]
+      }
+    },
+    "relative_action": {
+      "single_arm": {
+        "min": [
+          [
+            -14.407760620117188,
+            -27.423320770263672,
+            -21.879257202148438,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -14.407760620117188,
+            -27.423320770263672,
+            -22.60553741455078,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -14.407760620117188,
+            -27.423320770263672,
+            -25.817413330078125,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -15.768430709838867,
+            -27.423320770263672,
+            -29.261001586914062,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -17.888839721679688,
+            -27.423320770263672,
+            -32.529273986816406,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -19.93351936340332,
+            -27.423320770263672,
+            -35.610374450683594,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -21.82674217224121,
+            -27.68240737915039,
+            -38.79345703125,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -23.699081420898438,
+            -29.25457000732422,
+            -41.87455749511719,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -25.51657485961914,
+            -31.56486701965332,
+            -44.69236373901367,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -27.258338928222656,
+            -34.022499084472656,
+            -47.77346420288086,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -28.772918701171875,
+            -36.31063461303711,
+            -50.587806701660156,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -30.060306549072266,
+            -38.68351364135742,
+            -53.2205810546875,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -31.120513916015625,
+            -40.971649169921875,
+            -55.757965087890625,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -32.14179229736328,
+            -43.17504119873047,
+            -58.114105224609375,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -33.37945556640625,
+            -45.708072662353516,
+            -60.20216751098633,
+            -48.30026626586914,
+            -45.86080551147461
+          ],
+          [
+            -34.48583984375,
+            -48.25044631958008,
+            -62.19582748413086,
+            -48.30026626586914,
+            -45.86080551147461
+          ]
+        ],
+        "max": [
+          [
+            10.07719612121582,
+            18.55738067626953,
+            13.711788177490234,
+            11.358840942382812,
+            11.868131637573242
+          ],
+          [
+            12.046146392822266,
+            20.422576904296875,
+            13.711788177490234,
+            14.330146789550781,
+            14.163614273071289
+          ],
+          [
+            13.863639831542969,
+            22.11749267578125,
+            16.409061431884766,
+            16.994083404541016,
+            16.361417770385742
+          ],
+          [
+            15.792083740234375,
+            24.909435272216797,
+            19.24298095703125,
+            19.530155181884766,
+            18.168498992919922
+          ],
+          [
+            17.912492752075195,
+            28.13056182861328,
+            22.329681396484375,
+            22.153682708740234,
+            19.78022003173828
+          ],
+          [
+            19.7299861907959,
+            31.18191146850586,
+            25.320167541503906,
+            24.689754486083984,
+            21.098901748657227
+          ],
+          [
+            21.517221450805664,
+            34.232757568359375,
+            28.40127182006836,
+            27.05988311767578,
+            23.10134506225586
+          ],
+          [
+            23.25898551940918,
+            36.85987854003906,
+            31.215618133544922,
+            29.407485961914062,
+            25.10378646850586
+          ],
+          [
+            24.96111297607422,
+            39.402252197265625,
+            33.934242248535156,
+            31.68120574951172,
+            27.008548736572266
+          ],
+          [
+            26.32281494140625,
+            41.860923767089844,
+            36.5670166015625,
+            33.69257354736328,
+            28.913311004638672
+          ],
+          [
+            27.610206604003906,
+            44.233802795410156,
+            39.19501876831055,
+            35.63929748535156,
+            30.818073272705078
+          ],
+          [
+            28.746139526367188,
+            46.68543243408203,
+            41.737003326416016,
+            37.38831329345703,
+            32.625152587890625
+          ],
+          [
+            29.80634307861328,
+            49.737571716308594,
+            44.274383544921875,
+            38.92519760131836,
+            34.334556579589844
+          ],
+          [
+            30.57229995727539,
+            52.535362243652344,
+            46.634803771972656,
+            40.14950942993164,
+            36.141639709472656
+          ],
+          [
+            31.500513076782227,
+            55.078819274902344,
+            48.98831558227539,
+            40.76426315307617,
+            37.80220031738281
+          ],
+          [
+            33.01509094238281,
+            57.621192932128906,
+            51.16715621948242,
+            41.1302375793457,
+            39.120880126953125
+          ]
+        ],
+        "mean": [
+          [
+            0.10973429679870605,
+            -2.209338903427124,
+            -1.8722033500671387,
+            -0.15508916974067688,
+            -0.028145914897322655
+          ],
+          [
+            0.10968202352523804,
+            -2.2086832523345947,
+            -1.8730086088180542,
+            -0.154344841837883,
+            -0.028264271095395088
+          ],
+          [
+            0.10962218046188354,
+            -2.20805287361145,
+            -1.873792290687561,
+            -0.15358231961727142,
+            -0.028378551825881004
+          ],
+          [
+            0.10955725610256195,
+            -2.207461357116699,
+            -1.874544382095337,
+            -0.15280047059059143,
+            -0.02848830074071884
+          ],
+          [
+            0.10948893427848816,
+            -2.2069222927093506,
+            -1.8752955198287964,
+            -0.15200501680374146,
+            -0.02859322726726532
+          ],
+          [
+            0.10941897332668304,
+            -2.20639705657959,
+            -1.8760193586349487,
+            -0.15119512379169464,
+            -0.028693431988358498
+          ],
+          [
+            0.10934928804636002,
+            -2.2059144973754883,
+            -1.8767318725585938,
+            -0.1503802090883255,
+            -0.028788452968001366
+          ],
+          [
+            0.1092788428068161,
+            -2.205451726913452,
+            -1.8774136304855347,
+            -0.1495615541934967,
+            -0.02887848950922489
+          ],
+          [
+            0.109209343791008,
+            -2.2050089836120605,
+            -1.8780895471572876,
+            -0.1487341672182083,
+            -0.028965836390852928
+          ],
+          [
+            0.10913944989442825,
+            -2.2045910358428955,
+            -1.8787379264831543,
+            -0.14790278673171997,
+            -0.02905186079442501
+          ],
+          [
+            0.10906971991062164,
+            -2.2042009830474854,
+            -1.8793729543685913,
+            -0.14706464111804962,
+            -0.02913922630250454
+          ],
+          [
+            0.10899965465068817,
+            -2.2038326263427734,
+            -1.879947304725647,
+            -0.14621910452842712,
+            -0.029229290783405304
+          ],
+          [
+            0.10893117636442184,
+            -2.203528881072998,
+            -1.8804959058761597,
+            -0.14536546170711517,
+            -0.029323536902666092
+          ],
+          [
+            0.10886196792125702,
+            -2.2032315731048584,
+            -1.8810045719146729,
+            -0.14450465142726898,
+            -0.029420997947454453
+          ],
+          [
+            0.1087912917137146,
+            -2.2029733657836914,
+            -1.8814704418182373,
+            -0.1436382681131363,
+            -0.029520653188228607
+          ],
+          [
+            0.10871820151805878,
+            -2.2027506828308105,
+            -1.8818960189819336,
+            -0.14276832342147827,
+            -0.029622290283441544
+          ]
+        ],
+        "std": [
+          [
+            1.291955590248108,
+            2.895820140838623,
+            2.4284205436706543,
+            1.0855623483657837,
+            0.9884634017944336
+          ],
+          [
+            1.5667155981063843,
+            3.248924732208252,
+            2.7988362312316895,
+            1.2348947525024414,
+            1.1902235746383667
+          ],
+          [
+            1.8442904949188232,
+            3.62117600440979,
+            3.1847639083862305,
+            1.3908764123916626,
+            1.3953996896743774
+          ],
+          [
+            2.121694564819336,
+            4.004567623138428,
+            3.5781822204589844,
+            1.5495537519454956,
+            1.6011954545974731
+          ],
+          [
+            2.397252321243286,
+            4.3943281173706055,
+            3.9742770195007324,
+            1.7082926034927368,
+            1.8060038089752197
+          ],
+          [
+            2.6699206829071045,
+            4.787065505981445,
+            4.36962890625,
+            1.8654924631118774,
+            2.009084939956665
+          ],
+          [
+            2.9392967224121094,
+            5.180052757263184,
+            4.762381553649902,
+            2.0205588340759277,
+            2.209796190261841
+          ],
+          [
+            3.2050225734710693,
+            5.571920871734619,
+            5.15108060836792,
+            2.1726198196411133,
+            2.4078612327575684
+          ],
+          [
+            3.466845750808716,
+            5.961581707000732,
+            5.5352396965026855,
+            2.3211541175842285,
+            2.602936267852783
+          ],
+          [
+            3.724405527114868,
+            6.348147869110107,
+            5.91373348236084,
+            2.4658448696136475,
+            2.7950327396392822
+          ],
+          [
+            3.977760076522827,
+            6.730666160583496,
+            6.286655902862549,
+            2.606712579727173,
+            2.983889579772949
+          ],
+          [
+            4.226820468902588,
+            7.108645915985107,
+            6.653498649597168,
+            2.7434158325195312,
+            3.169560194015503
+          ],
+          [
+            4.471673965454102,
+            7.482038974761963,
+            7.014378070831299,
+            2.8758957386016846,
+            3.3519320487976074
+          ],
+          [
+            4.7121500968933105,
+            7.85037899017334,
+            7.368745803833008,
+            3.004105567932129,
+            3.5309762954711914
+          ],
+          [
+            4.9483323097229,
+            8.213367462158203,
+            7.716771602630615,
+            3.1281301975250244,
+            3.7066855430603027
+          ],
+          [
+            5.179955005645752,
+            8.571035385131836,
+            8.058523178100586,
+            3.248170852661133,
+            3.879103422164917
+          ]
+        ],
+        "q01": [
+          [
+            -3.880250930786133,
+            -8.599881172180176,
+            -9.154699325561523,
+            -3.436232089996338,
+            -3.425642251968384
+          ],
+          [
+            -4.857089996337891,
+            -9.876320838928223,
+            -10.865798950195312,
+            -3.8265511989593506,
+            -4.346764087677002
+          ],
+          [
+            -5.876626968383789,
+            -11.200237274169922,
+            -12.55186939239502,
+            -4.3084259033203125,
+            -5.274723052978516
+          ],
+          [
+            -6.864691734313965,
+            -12.497199058532715,
+            -14.240229606628418,
+            -4.79156494140625,
+            -6.153846263885498
+          ],
+          [
+            -7.865577220916748,
+            -13.833037376403809,
+            -15.899080276489258,
+            -5.358405590057373,
+            -7.0329670906066895
+          ],
+          [
+            -8.856843948364258,
+            -15.103858947753906,
+            -17.418010711669922,
+            -5.889519214630127,
+            -7.912088394165039
+          ],
+          [
+            -9.819734573364258,
+            -16.37712860107422,
+            -18.992219924926758,
+            -6.4837646484375,
+            -8.791208267211914
+          ],
+          [
+            -10.775996208190918,
+            -17.66802978515625,
+            -20.44565200805664,
+            -6.9591569900512695,
+            -9.670330047607422
+          ],
+          [
+            -11.732026100158691,
+            -18.8674259185791,
+            -21.919971466064453,
+            -7.49139928817749,
+            -10.556291580200195
+          ],
+          [
+            -12.65311050415039,
+            -20.0970516204834,
+            -23.280298233032227,
+            -7.985642910003662,
+            -11.428571701049805
+          ],
+          [
+            -13.55053997039795,
+            -21.310543060302734,
+            -24.66961669921875,
+            -8.45219612121582,
+            -12.258852005004883
+          ],
+          [
+            -14.445900917053223,
+            -22.47014617919922,
+            -25.971336364746094,
+            -8.819541931152344,
+            -13.089134216308594
+          ],
+          [
+            -15.32150650024414,
+            -23.607933044433594,
+            -27.2856502532959,
+            -9.188517570495605,
+            -13.919414520263672
+          ],
+          [
+            -16.171899795532227,
+            -24.695764541625977,
+            -28.55253028869629,
+            -9.590324401855469,
+            -14.700855255126953
+          ],
+          [
+            -17.013334274291992,
+            -25.719482421875,
+            -29.74740219116211,
+            -9.954161643981934,
+            -15.531135559082031
+          ],
+          [
+            -17.83913803100586,
+            -26.79294204711914,
+            -30.886316299438477,
+            -10.289176940917969,
+            -16.263734817504883
+          ]
+        ],
+        "q99": [
+          [
+            4.964918613433838,
+            5.23189640045166,
+            5.2053070068359375,
+            2.6407511234283447,
+            3.9072036743164062
+          ],
+          [
+            6.182013511657715,
+            6.905364513397217,
+            6.732197284698486,
+            3.2062454223632812,
+            4.9328460693359375
+          ],
+          [
+            7.405695915222168,
+            8.520384788513184,
+            8.360859870910645,
+            3.5569162368774414,
+            5.958484649658203
+          ],
+          [
+            8.610760688781738,
+            10.048929214477539,
+            9.90613842010498,
+            4.091148376464844,
+            6.89331579208374
+          ],
+          [
+            9.801132202148438,
+            11.651291847229004,
+            11.436395645141602,
+            4.566997051239014,
+            7.863248348236084
+          ],
+          [
+            10.974348068237305,
+            13.181883811950684,
+            12.88714599609375,
+            5.0329437255859375,
+            8.79120922088623
+          ],
+          [
+            12.129141807556152,
+            14.657190322875977,
+            14.390020370483398,
+            5.57182502746582,
+            9.768009185791016
+          ],
+          [
+            13.220808029174805,
+            16.158241271972656,
+            15.842169761657715,
+            6.1076812744140625,
+            10.647130966186523
+          ],
+          [
+            14.329455375671387,
+            17.674537658691406,
+            17.28555679321289,
+            6.6834516525268555,
+            11.526250839233398
+          ],
+          [
+            15.360580444335938,
+            19.114492416381836,
+            18.61423110961914,
+            7.178110599517822,
+            12.45421314239502
+          ],
+          [
+            16.411075592041016,
+            20.48096466064453,
+            19.966537475585938,
+            7.684507846832275,
+            13.333332061767578
+          ],
+          [
+            17.393070220947266,
+            21.859176635742188,
+            21.3115291595459,
+            8.17349910736084,
+            14.261293411254883
+          ],
+          [
+            18.34865951538086,
+            23.18575096130371,
+            22.599977493286133,
+            8.620201110839844,
+            15.042734146118164
+          ],
+          [
+            19.292339324951172,
+            24.476245880126953,
+            23.886260986328125,
+            8.993049621582031,
+            15.921854972839355
+          ],
+          [
+            20.147075653076172,
+            25.74358367919922,
+            25.125085830688477,
+            9.504680633544922,
+            16.703296661376953
+          ],
+          [
+            21.00503158569336,
+            26.99942398071289,
+            26.240026473999023,
+            9.915670394897461,
+            17.53357696533203
+          ]
+        ]
+      }
+    }
+  }
+}

checkpoint-10000/experiment_cfg/final_model_config.json

@@ -0,0 +1,54 @@
+{
+  "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": false,
+  "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,
+  "extra_augmentation_config": null,
+  "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
+}

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

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

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

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

checkpoint-10000/optimizer.pt

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

checkpoint-10000/processor_config.json

@@ -0,0 +1,456 @@
+{
+  "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": [
+            "front",
+            "wrist_color",
+            "wrist_depth"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": null
+        },
+        "state": {
+          "delta_indices": [
+            0
+          ],
+          "modality_keys": [
+            "single_arm",
+            "gripper"
+          ],
+          "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": [
+            "single_arm",
+            "gripper"
+          ],
+          "sin_cos_embedding_keys": null,
+          "mean_std_embedding_keys": null,
+          "action_configs": [
+            {
+              "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": [
+            "annotation.human.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
+  }
+}

checkpoint-10000/rng_state.pth

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

checkpoint-10000/scheduler.pt

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

checkpoint-10000/training_args.bin

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

checkpoint-10000/wandb_config.json

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