Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,13 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+19999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+29999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+39999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+49999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+59999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+69999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+79999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+89999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+9999/default/checkpoint filter=lfs diff=lfs merge=lfs -text
+99999/default/checkpoint filter=lfs diff=lfs merge=lfs -text

19999/default/checkpoint

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

29999/default/checkpoint

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

39999/default/checkpoint

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

49999/default/checkpoint

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

59999/default/checkpoint

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

69999/default/checkpoint

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

79999/default/checkpoint

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

89999/default/checkpoint

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

9999/default/checkpoint

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

99999/default/checkpoint

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

config.json

@@ -0,0 +1 @@
+{"seed": 42, "num_steps": 300000, "save_dir": null, "model": {"observation_tokenizers": {"primary": {"module": "octo.model.components.tokenizers", "name": "ImageTokenizer", "args": [], "kwargs": {"obs_stack_keys": ["image_primary"], "task_stack_keys": ["image_primary"], "encoder": {"module": "octo.model.components.vit_encoders", "name": "SmallStem16", "args": [], "kwargs": {}}}}}, "task_tokenizers": {"language": {"module": "octo.model.components.tokenizers", "name": "LanguageTokenizer", "args": [], "kwargs": {"encoder": "t5-base", "finetune_encoder": false}}}, "heads": {"action": {"module": "octo.model.components.action_heads", "name": "DiffusionActionHead", "args": [], "kwargs": {"readout_key": "readout_action", "use_map": false, "action_dim": 7, "action_horizon": 4}}}, "readouts": {"action": 1}, "token_embedding_size": 384, "transformer_kwargs": {"attention_dropout_rate": 0.0, "add_position_embedding": false, "num_layers": 12, "mlp_dim": 1536, "num_attention_heads": 6, "dropout_rate": 0.0}, "max_horizon": 10}, "window_size": 2, "dataset_kwargs": {"oxe_kwargs": {"data_mix": "oxe_magic_soup", "data_dir": "gs://rail-octo-central2/resize_256_256", "load_camera_views": ["primary", "wrist"], "load_depth": false}, "traj_transform_kwargs": {"window_size": 2, "future_action_window_size": 3, "goal_relabeling_strategy": "uniform", "subsample_length": 100, "task_augment_strategy": "delete_task_conditioning", "task_augment_kwargs": {"keep_image_prob": 0.5}}, "frame_transform_kwargs": {"num_parallel_calls": 200, "resize_size": {"primary": [256, 256], "wrist": [128, 128]}, "image_augment_kwargs": [{"random_resized_crop": {"scale": [0.8, 1.0], "ratio": [0.9, 1.1]}, "random_brightness": [0.1], "random_contrast": [0.9, 1.1], "random_saturation": [0.9, 1.1], "random_hue": [0.05], "augment_order": ["random_resized_crop", "random_brightness", "random_contrast", "random_saturation", "random_hue"]}, {"random_brightness": [0.1], "random_contrast": [0.9, 1.1], "random_saturation": [0.9, 1.1], "random_hue": [0.05], "augment_order": ["random_brightness", "random_contrast", "random_saturation", "random_hue"]}]}, "traj_transform_threads": 48, "traj_read_threads": 48, "shuffle_buffer_size": 500000, "batch_size": 128, "balance_weights": true}, "optimizer": {"learning_rate": {"name": "rsqrt", "init_value": 0.0, "peak_value": 0.0003, "warmup_steps": 2000, "timescale": 10000}, "weight_decay": 0.1, "clip_gradient": 1.0, "frozen_keys": ["*hf_model*"]}, "prefetch_num_batches": 0, "start_step": null, "log_interval": 100, "eval_interval": 5000, "viz_interval": 20000, "save_interval": 10000, "val_kwargs": {"val_shuffle_buffer_size": 1000, "num_val_batches": 16}, "viz_kwargs": {"eval_batch_size": 128, "trajs_for_metrics": 100, "trajs_for_viz": 8, "samples_per_state": 8}, "resume_path": null, "text_processor": {"module": "octo.data.utils.text_processing", "name": "HFTokenizer", "args": [], "kwargs": {"encode_with_model": false, "tokenizer_kwargs": {"max_length": 16, "padding": "max_length", "truncation": true, "return_tensors": "np"}, "tokenizer_name": "t5-base"}}, "pretrained_loaders": [{"module": "octo.utils.train_utils", "name": "hf_weights_loader", "args": [], "kwargs": {"hf_model": "t5-base"}}], "wandb": {"project": "octo", "group": null, "entity": null}, "wandb_resume_id": null, "eval_datasets": ["bridge_dataset"]}

dataset_statistics.json

@@ -0,0 +1 @@
+{"action": {"max": [0.12335848808288574, 0.14755941927433014, 0.12199220061302185, 3.1415927410125732, 1.5704798698425293, 3.141592502593994, 1.0], "mean": [-0.0025376665871590376, 0.0001923212403198704, -0.0007950640865601599, -0.0826713889837265, -0.11953045427799225, -0.08652631938457489, 0.6203598976135254], "min": [-0.1378057897090912, -0.14709188044071198, -0.1626778244972229, -3.1415927410125732, -1.5705456733703613, -3.141592264175415, 0.0], "p01": [-0.0655677542090416, -0.054904356598854065, -0.06744533777236938, -3.1395938396453857, -1.292111873626709, -3.07131028175354, 0.0], "p99": [0.06519947946071625, 0.0555696003139019, 0.053435586392879486, 3.1396241188049316, 0.9427980780601501, 3.0678741931915283, 1.0], "std": [0.02172505296766758, 0.017638077959418297, 0.020821034908294678, 3.0158021450042725, 0.4210449755191803, 1.4313385486602783, 0.4551418125629425]}, "num_trajectories": 92233, "num_transitions": 27044326}

example_batch.msgpack

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