viis0002 commited on 29 days ago

Commit

d77f03e

verified ·

1 Parent(s): 462d4bd

Upload folder using huggingface_hub

Browse files

Files changed (18) hide show

config.json +70 -0
embodiment_id.json +11 -0
experiment_cfg/conf.yaml +206 -0
experiment_cfg/config.yaml +240 -0
experiment_cfg/dataset_statistics.json +824 -0
experiment_cfg/final_model_config.json +54 -0
experiment_cfg/final_processor_config.json +0 -0
model-00001-of-00002.safetensors +3 -0
model-00002-of-00002.safetensors +3 -0
model.safetensors.index.json +0 -0
optimizer.pt +3 -0
processor_config.json +455 -0
rng_state.pth +3 -0
scheduler.pt +3 -0
statistics.json +0 -0
trainer_state.json +0 -0
training_args.bin +3 -0
wandb_config.json +1 -0

config.json ADDED Viewed

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

embodiment_id.json ADDED Viewed

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

experiment_cfg/conf.yaml ADDED Viewed

	@@ -0,0 +1,206 @@

+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: null
+  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/ps/viis0002/cluttered_picking_bin
+    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
+        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_RGB
+  experiment_name: null
+  max_steps: 10000
+  global_batch_size: 32
+  batch_size: null
+  gradient_accumulation_steps: 1
+  learning_rate: 0.0001
+  lr_scheduler_type: cosine
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.05
+  warmup_steps: 0
+  max_grad_norm: 1.0
+  optim: adamw_torch
+  start_from_checkpoint: nvidia/GR00T-N1.6-3B
+  tf32: true
+  fp16: false
+  bf16: true
+  eval_bf16: true
+  logging_steps: 10
+  save_steps: 4000
+  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: 4
+  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: 4000

experiment_cfg/config.yaml ADDED Viewed

	@@ -0,0 +1,240 @@

+!!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/ps/viis0002/cluttered_picking_bin
+    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
+        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: null
+  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: 4
+  ddp_bucket_cap_mb: 100
+  deepspeed_stage: 2
+  enable_open_loop_eval: false
+  enable_profiling: false
+  eval_batch_size: 2
+  eval_bf16: true
+  eval_set_split_ratio: 0.1
+  eval_steps: 500
+  eval_strategy: 'no'
+  experiment_name: null
+  fp16: false
+  global_batch_size: 32
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: false
+  learning_rate: 0.0001
+  logging_steps: 10
+  lr_scheduler_type: cosine
+  max_concurrent_uploads: 2
+  max_grad_norm: 1.0
+  max_retries: 3
+  max_steps: 10000
+  num_gpus: 1
+  open_loop_eval_plot_indices: null
+  open_loop_eval_steps_per_traj: 100
+  open_loop_eval_traj_ids:
+  - 0
+  optim: adamw_torch
+  output_dir: ./output_RGB
+  remove_unused_columns: false
+  save_best_eval_metric_greater_is_better: true
+  save_best_eval_metric_name: ''
+  save_steps: 4000
+  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

experiment_cfg/dataset_statistics.json ADDED Viewed

	@@ -0,0 +1,824 @@

+{
+  "new_embodiment": {
+    "state": {
+      "single_arm": {
+        "min": [
+          -39.36348342895508,
+          -99.65782928466797,
+          -61.97892761230469,
+          55.84988784790039,
+          -26.398046493530273
+        ],
+        "max": [
+          7.956449031829834,
+          49.1017951965332,
+          100.0,
+          99.55850219726562,
+          57.851036071777344
+        ],
+        "mean": [
+          -6.385418343874992,
+          -24.622329268110402,
+          18.56802663314117,
+          81.77844942661774,
+          10.85042606146218
+        ],
+        "std": [
+          10.46960328595638,
+          51.62350175979786,
+          53.14153035933821,
+          10.531352359119015,
+          18.254572925078886
+        ],
+        "q01": [
+          -28.374080331028793,
+          -98.99477709610561,
+          -29.833800342347036,
+          66.12401334198283,
+          -1.0385594413972399
+        ],
+        "q99": [
+          1.9539289179059653,
+          34.71050091958405,
+          96.99906607704047,
+          93.29626893615101,
+          23.499264058127626
+        ]
+      },
+      "gripper": {
+        "min": [
+          0.8333333134651184
+        ],
+        "max": [
+          68.52564239501953
+        ],
+        "mean": [
+          8.585488289392284
+        ],
+        "std": [
+          11.661403495903766
+        ],
+        "q01": [
+          0.966566352866377
+        ],
+        "q99": [
+          42.39703828250058
+        ]
+      }
+    },
+    "action": {
+      "single_arm": {
+        "min": [
+          -39.73866271972656,
+          -100.0,
+          -65.7715072631836,
+          54.796321868896484,
+          -26.886445999145508
+        ],
+        "max": [
+          8.685626029968262,
+          48.8510627746582,
+          97.72416687011719,
+          100.0,
+          58.19291687011719
+        ],
+        "mean": [
+          -6.205533711700667,
+          -26.562827958046608,
+          16.34602250075434,
+          81.57096115074431,
+          10.85811976397703
+        ],
+        "std": [
+          10.59422603504957,
+          50.802737184612496,
+          53.54985182039301,
+          10.726019282916475,
+          18.349091877543735
+        ],
+        "q01": [
+          -28.653126963685672,
+          -99.79199853840126,
+          -32.71377093241167,
+          65.14824963891643,
+          -1.275976052805534
+        ],
+        "q99": [
+          2.4851899093151366,
+          34.353396852719165,
+          94.1144943032115,
+          93.66894998114458,
+          23.70856088634366
+        ]
+      },
+      "gripper": {
+        "min": [
+          0.0
+        ],
+        "max": [
+          68.81810760498047
+        ],
+        "mean": [
+          6.1142540385448525
+        ],
+        "std": [
+          11.503861267012631
+        ],
+        "q01": [
+          0.4977736616828879
+        ],
+        "q99": [
+          42.733437176746975
+        ]
+      }
+    },
+    "relative_action": {
+      "single_arm": {
+        "min": [
+          [
+            -4.869014739990234,
+            -15.571052551269531,
+            -15.518726348876953,
+            -7.186775207519531,
+            -7.228327751159668
+          ],
+          [
+            -6.329431533813477,
+            -15.571052551269531,
+            -18.816944122314453,
+            -9.114067077636719,
+            -8.98656940460205
+          ],
+          [
+            -7.716586112976074,
+            -16.198678970336914,
+            -22.023540496826172,
+            -10.880073547363281,
+            -10.647130966186523
+          ],
+          [
+            -9.177002906799316,
+            -18.16617202758789,
+            -25.13851547241211,
+            -12.646080017089844,
+            -12.45421028137207
+          ],
+          [
+            -10.517036437988281,
+            -20.20210075378418,
+            -28.233642578125,
+            -14.398155212402344,
+            -14.261293411254883
+          ],
+          [
+            -11.77331829071045,
+            -22.31884002685547,
+            -31.257003784179688,
+            -16.07585906982422,
+            -16.0195369720459
+          ],
+          [
+            -13.053546905517578,
+            -24.457420349121094,
+            -34.097137451171875,
+            -17.665260314941406,
+            -17.77777862548828
+          ],
+          [
+            -14.309828758239746,
+            -26.585079193115234,
+            -36.91917037963867,
+            -19.066925048828125,
+            -19.53601837158203
+          ],
+          [
+            -15.539650917053223,
+            -28.712738037109375,
+            -39.7411994934082,
+            -20.303131103515625,
+            -21.294261932373047
+          ],
+          [
+            -16.615745544433594,
+            -30.84039878845215,
+            -42.39809036254883,
+            -21.529586791992188,
+            -22.905982971191406
+          ],
+          [
+            -17.69184112548828,
+            -32.904354095458984,
+            -45.146602630615234,
+            -22.677490234375,
+            -24.46886444091797
+          ],
+          [
+            -18.696866989135742,
+            -35.03201675415039,
+            -47.786563873291016,
+            -23.737091064453125,
+            -26.178268432617188
+          ],
+          [
+            -19.619234085083008,
+            -37.085052490234375,
+            -50.33549499511719,
+            -24.613136291503906,
+            -27.74114990234375
+          ],
+          [
+            -20.541601181030273,
+            -39.117122650146484,
+            -53.05598449707031,
+            -25.4078369140625,
+            -29.206348419189453
+          ],
+          [
+            -21.462875366210938,
+            -41.17015838623047,
+            -55.7408561706543,
+            -26.10866928100586,
+            -30.76923179626465
+          ],
+          [
+            -22.30837631225586,
+            -43.23316955566406,
+            -58.580989837646484,
+            -26.72677230834961,
+            -32.23442840576172
+          ]
+        ],
+        "max": [
+          [
+            6.400243759155273,
+            12.026496887207031,
+            10.099809646606445,
+            7.688037872314453,
+            9.670330047607422
+          ],
+          [
+            8.159036636352539,
+            15.023571014404297,
+            13.306406021118164,
+            8.936538696289062,
+            12.014652252197266
+          ],
+          [
+            9.77318000793457,
+            18.217063903808594,
+            16.51300048828125,
+            10.86383056640625,
+            14.35897445678711
+          ],
+          [
+            11.425369262695312,
+            21.28089141845703,
+            19.627979278564453,
+            12.718132019042969,
+            16.75213623046875
+          ],
+          [
+            13.100410461425781,
+            24.27490234375,
+            22.72311019897461,
+            14.484138488769531,
+            19.047618865966797
+          ],
+          [
+            14.714553833007812,
+            27.253623962402344,
+            25.818239212036133,
+            16.236221313476562,
+            21.24542236328125
+          ],
+          [
+            16.222091674804688,
+            30.147239685058594,
+            28.913368225097656,
+            17.913925170898438,
+            23.394384384155273
+          ],
+          [
+            17.682506561279297,
+            33.04085922241211,
+            32.00849914550781,
+            19.40319061279297,
+            25.738704681396484
+          ],
+          [
+            19.106292724609375,
+            35.849369049072266,
+            35.031864166259766,
+            20.727691650390625,
+            28.083026885986328
+          ],
+          [
+            20.446325302124023,
+            38.5012092590332,
+            38.05522537231445,
+            21.875595092773438,
+            30.28083038330078
+          ],
+          [
+            21.753013610839844,
+            41.277400970458984,
+            41.059322357177734,
+            23.014450073242188,
+            32.42979431152344
+          ],
+          [
+            23.05970001220703,
+            43.92923355102539,
+            43.97238540649414,
+            24.074050903320312,
+            34.57875442504883
+          ],
+          [
+            24.232229232788086,
+            46.58107376098633,
+            46.97648239135742,
+            25.12529754638672,
+            36.43467712402344
+          ],
+          [
+            25.321006774902344,
+            49.2193717956543,
+            49.88954544067383,
+            26.00133514404297,
+            38.43711853027344
+          ],
+          [
+            26.32023811340332,
+            51.857669830322266,
+            52.802608489990234,
+            26.807876586914062,
+            40.29304122924805
+          ],
+          [
+            27.319469451904297,
+            54.32575225830078,
+            55.806705474853516,
+            27.771522521972656,
+            41.904762268066406
+          ]
+        ],
+        "mean": [
+          [
+            0.1785251498222351,
+            -1.9557442665100098,
+            -2.2593743801116943,
+            -0.20868544280529022,
+            0.008650688454508781
+          ],
+          [
+            0.1787870079278946,
+            -1.9535331726074219,
+            -2.2614686489105225,
+            -0.20898747444152832,
+            0.008710959926247597
+          ],
+          [
+            0.17905102670192719,
+            -1.9514031410217285,
+            -2.2634711265563965,
+            -0.20927496254444122,
+            0.008769187144935131
+          ],
+          [
+            0.17931483685970306,
+            -1.9493659734725952,
+            -2.265377998352051,
+            -0.20954172313213348,
+            0.008825326338410378
+          ],
+          [
+            0.1795794814825058,
+            -1.947402000427246,
+            -2.26723575592041,
+            -0.20978474617004395,
+            0.008879491128027439
+          ],
+          [
+            0.17984697222709656,
+            -1.945528268814087,
+            -2.2690117359161377,
+            -0.20999972522258759,
+            0.008933082222938538
+          ],
+          [
+            0.18011674284934998,
+            -1.9437333345413208,
+            -2.2707107067108154,
+            -0.2101903259754181,
+            0.008988128043711185
+          ],
+          [
+            0.18038682639598846,
+            -1.942039132118225,
+            -2.2723309993743896,
+            -0.21035927534103394,
+            0.009043456986546516
+          ],
+          [
+            0.18065497279167175,
+            -1.9404367208480835,
+            -2.2738921642303467,
+            -0.21050825715065002,
+            0.009101650677621365
+          ],
+          [
+            0.18092294037342072,
+            -1.9389110803604126,
+            -2.275402545928955,
+            -0.21062998473644257,
+            0.009162607602775097
+          ],
+          [
+            0.18119066953659058,
+            -1.937473177909851,
+            -2.2768537998199463,
+            -0.210730642080307,
+            0.009221411310136318
+          ],
+          [
+            0.18145936727523804,
+            -1.9361056089401245,
+            -2.2782578468322754,
+            -0.21082045137882233,
+            0.009277415461838245
+          ],
+          [
+            0.18173053860664368,
+            -1.9348115921020508,
+            -2.2796051502227783,
+            -0.21090328693389893,
+            0.00933289248496294
+          ],
+          [
+            0.18200388550758362,
+            -1.9335931539535522,
+            -2.280888795852661,
+            -0.21098707616329193,
+            0.009389038197696209
+          ],
+          [
+            0.18227829039096832,
+            -1.9324415922164917,
+            -2.2821178436279297,
+            -0.2110622674226761,
+            0.009444031864404678
+          ],
+          [
+            0.18255291879177094,
+            -1.9313602447509766,
+            -2.2832837104797363,
+            -0.21112234890460968,
+            0.009497801773250103
+          ]
+        ],
+        "std": [
+          [
+            0.9654352068901062,
+            3.4649360179901123,
+            3.150923252105713,
+            1.1157993078231812,
+            1.0172107219696045
+          ],
+          [
+            1.1931160688400269,
+            4.058739185333252,
+            3.8396174907684326,
+            1.3442760705947876,
+            1.2842698097229004
+          ],
+          [
+            1.4228914976119995,
+            4.675368785858154,
+            4.540013790130615,
+            1.574043869972229,
+            1.5493543148040771
+          ],
+          [
+            1.6525599956512451,
+            5.304614543914795,
+            5.245314598083496,
+            1.8014473915100098,
+            1.8111250400543213
+          ],
+          [
+            1.8808844089508057,
+            5.940498352050781,
+            5.951900005340576,
+            2.0248169898986816,
+            2.0692081451416016
+          ],
+          [
+            2.1071860790252686,
+            6.579427242279053,
+            6.656661510467529,
+            2.243411064147949,
+            2.323486804962158
+          ],
+          [
+            2.331037759780884,
+            7.219012260437012,
+            7.358837604522705,
+            2.4567863941192627,
+            2.5739219188690186
+          ],
+          [
+            2.552251100540161,
+            7.857221603393555,
+            8.057107925415039,
+            2.664818286895752,
+            2.8202738761901855
+          ],
+          [
+            2.7706377506256104,
+            8.493036270141602,
+            8.75144100189209,
+            2.8674299716949463,
+            3.0627496242523193
+          ],
+          [
+            2.986109733581543,
+            9.125910758972168,
+            9.44082260131836,
+            3.064739465713501,
+            3.301175355911255
+          ],
+          [
+            3.198592185974121,
+            9.754792213439941,
+            10.125152587890625,
+            3.2569146156311035,
+            3.535595178604126
+          ],
+          [
+            3.408093214035034,
+            10.379477500915527,
+            10.80418872833252,
+            3.443956136703491,
+            3.7658936977386475
+          ],
+          [
+            3.614506959915161,
+            10.999598503112793,
+            11.477699279785156,
+            3.6260931491851807,
+            3.992025852203369
+          ],
+          [
+            3.8178415298461914,
+            11.614961624145508,
+            12.145866394042969,
+            3.8034627437591553,
+            4.214049816131592
+          ],
+          [
+            4.01806116104126,
+            12.225295066833496,
+            12.80855655670166,
+            3.9761598110198975,
+            4.431927680969238
+          ],
+          [
+            4.215169906616211,
+            12.830470085144043,
+            13.465753555297852,
+            4.144509792327881,
+            4.6458001136779785
+          ]
+        ],
+        "q01": [
+          [
+            -3.1989338397979736,
+            -9.316668510437012,
+            -11.341864585876465,
+            -3.8126816749572754,
+            -3.5653228759765625
+          ],
+          [
+            -4.159060478210449,
+            -10.740071296691895,
+            -13.675247192382812,
+            -4.741844177246094,
+            -4.590963363647461
+          ],
+          [
+            -5.090120315551758,
+            -12.214762687683105,
+            -15.99472713470459,
+            -5.634047508239746,
+            -5.612699031829834
+          ],
+          [
+            -6.020403861999512,
+            -13.759285926818848,
+            -18.29571533203125,
+            -6.501101493835449,
+            -6.593406677246094
+          ],
+          [
+            -6.948093891143799,
+            -15.280621528625488,
+            -20.582651138305664,
+            -7.327765464782715,
+            -7.5702056884765625
+          ],
+          [
+            -7.856415271759033,
+            -16.814342498779297,
+            -22.871421813964844,
+            -8.132277488708496,
+            -8.4981689453125
+          ],
+          [
+            -8.780505180358887,
+            -18.380815505981445,
+            -25.133073806762695,
+            -8.922734260559082,
+            -9.474967956542969
+          ],
+          [
+            -9.661839485168457,
+            -19.964109420776367,
+            -27.354209899902344,
+            -9.681880950927734,
+            -10.35409164428711
+          ],
+          [
+            -10.548376083374023,
+            -21.490734100341797,
+            -29.52994728088379,
+            -10.417035102844238,
+            -11.233211517333984
+          ],
+          [
+            -11.400496482849121,
+            -23.03583335876465,
+            -31.718549728393555,
+            -11.105175971984863,
+            -12.161170959472656
+          ],
+          [
+            -12.23694896697998,
+            -24.55828094482422,
+            -33.89005661010742,
+            -11.810083389282227,
+            -13.085226058959961
+          ],
+          [
+            -13.08798599243164,
+            -26.0482120513916,
+            -36.02480697631836,
+            -12.4576416015625,
+            -13.968254089355469
+          ],
+          [
+            -13.868152618408203,
+            -27.571319580078125,
+            -38.08620071411133,
+            -13.088216781616211,
+            -14.847375869750977
+          ],
+          [
+            -14.652140617370605,
+            -29.0412540435791,
+            -40.186424255371094,
+            -13.666051864624023,
+            -15.775335311889648
+          ],
+          [
+            -15.41779899597168,
+            -30.49142837524414,
+            -42.280601501464844,
+            -14.259786605834961,
+            -16.60561752319336
+          ],
+          [
+            -16.16594886779785,
+            -31.97154426574707,
+            -44.370018005371094,
+            -14.746328353881836,
+            -17.484739303588867
+          ]
+        ],
+        "q99": [
+          [
+            3.2479984760284424,
+            7.912990093231201,
+            5.633664608001709,
+            3.4832823276519775,
+            5.372402191162109
+          ],
+          [
+            4.097276210784912,
+            10.157513618469238,
+            7.765993118286133,
+            4.481320381164551,
+            6.837606430053711
+          ],
+          [
+            4.946655750274658,
+            12.368200302124023,
+            9.88430404663086,
+            5.5072197914123535,
+            8.298991203308105
+          ],
+          [
+            5.767092704772949,
+            14.587282180786133,
+            11.932137489318848,
+            6.434570789337158,
+            9.715353965759277
+          ],
+          [
+            6.5880303382873535,
+            16.77672004699707,
+            14.06669807434082,
+            7.37797737121582,
+            11.037851333618164
+          ],
+          [
+            7.4164137840271,
+            18.925695419311523,
+            16.074583053588867,
+            8.235647201538086,
+            12.454212188720703
+          ],
+          [
+            8.199186325073242,
+            21.071264266967773,
+            18.116865158081055,
+            9.053117752075195,
+            13.724054336547852
+          ],
+          [
+            8.967052459716797,
+            23.139829635620117,
+            20.07178497314453,
+            9.901945114135742,
+            15.087759971618652
+          ],
+          [
+            9.712544441223145,
+            25.245664596557617,
+            22.007034301757812,
+            10.643431663513184,
+            16.21489715576172
+          ],
+          [
+            10.44395923614502,
+            27.225433349609375,
+            23.908384323120117,
+            11.467199325561523,
+            17.484737396240234
+          ],
+          [
+            11.168615341186523,
+            29.31740951538086,
+            25.754091262817383,
+            12.12663745880127,
+            18.65690040588379
+          ],
+          [
+            11.87579345703125,
+            31.247526168823242,
+            27.57501983642578,
+            12.88122844696045,
+            19.776405334472656
+          ],
+          [
+            12.570459365844727,
+            33.20111846923828,
+            29.390729904174805,
+            13.581562042236328,
+            20.854700088500977
+          ],
+          [
+            13.26676082611084,
+            35.10786819458008,
+            31.10728645324707,
+            14.265336036682129,
+            21.733821868896484
+          ],
+          [
+            13.936410903930664,
+            36.96277618408203,
+            32.90301513671875,
+            14.87486743927002,
+            22.905981063842773
+          ],
+          [
+            14.615703582763672,
+            38.77449417114258,
+            34.64031219482422,
+            15.497631072998047,
+            23.78510284423828
+          ]
+        ]
+      }
+    }
+  }
+}

experiment_cfg/final_model_config.json ADDED Viewed

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

experiment_cfg/final_processor_config.json ADDED Viewed

The diff for this file is too large to render. See raw diff

model-00001-of-00002.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:f657b16c32e74a5caa3f1b985d0466c1dd06b7ed39adcfbb26c9d967975859c5
+size 4990120184

model-00002-of-00002.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e91421ab3c69ef8eee1dbfc65b30a3143f24abea48061c1cb7a75abd4bcf82a8
+size 4823190320

model.safetensors.index.json ADDED Viewed

The diff for this file is too large to render. See raw diff

optimizer.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:2976e4c31a933c757f52bd6117ccc388d4855e22a11929d7831a1e1aea76c310
+size 12960193762

processor_config.json ADDED Viewed

	@@ -0,0 +1,455 @@

+{
+  "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"
+          ],
+          "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
+  }
+}

rng_state.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:20a0c565bd08e39a00dedd5287e1c398c93d4b9bfb3068b49a86d6103c37e0a0
+size 14645

scheduler.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:de7de3f59b6396d7d4468481ee0f0884cb3ddcb21a44cf41b6fd5a4c73ee0149
+size 1465

statistics.json ADDED Viewed

The diff for this file is too large to render. See raw diff

trainer_state.json ADDED Viewed

The diff for this file is too large to render. See raw diff

training_args.bin ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:8ccdc4abdd7c9589099fa367b7addfc5f3480910fc3ed6e9c5a6f87301789adc
+size 5713

wandb_config.json ADDED Viewed

	@@ -0,0 +1 @@


1	+ {"project": "finetune-gr00t-n1d6", "run_id": "output_RGB"}