Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ 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
+wandb/run-20250612_232715-sam_single_cube/run-sam_single_cube.wandb filter=lfs diff=lfs merge=lfs -text

checkpoints/000600/pretrained_model/config.json

@@ -0,0 +1,72 @@
+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                7
+            ]
+        },
+        "observation.images.top_camera": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.bottom_camera": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.right": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                7
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoints/000600/pretrained_model/model.safetensors

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

checkpoints/000600/pretrained_model/train_config.json

@@ -0,0 +1,178 @@
+{
+    "dataset": {
+        "repo_id": "girardijp/sam_pick_cube_single",
+        "root": null,
+        "episodes": null,
+        "image_transforms": {
+            "enable": false,
+            "max_num_transforms": 3,
+            "random_order": false,
+            "tfs": {
+                "brightness": {
+                    "weight": 1.0,
+                    "type": "ColorJitter",
+                    "kwargs": {
+                        "brightness": [
+                            0.8,
+                            1.2
+                        ]
+                    }
+                },
+                "contrast": {
+                    "weight": 1.0,
+                    "type": "ColorJitter",
+                    "kwargs": {
+                        "contrast": [
+                            0.8,
+                            1.2
+                        ]
+                    }
+                },
+                "saturation": {
+                    "weight": 1.0,
+                    "type": "ColorJitter",
+                    "kwargs": {
+                        "saturation": [
+                            0.5,
+                            1.5
+                        ]
+                    }
+                },
+                "hue": {
+                    "weight": 1.0,
+                    "type": "ColorJitter",
+                    "kwargs": {
+                        "hue": [
+                            -0.05,
+                            0.05
+                        ]
+                    }
+                },
+                "sharpness": {
+                    "weight": 1.0,
+                    "type": "SharpnessJitter",
+                    "kwargs": {
+                        "sharpness": [
+                            0.5,
+                            1.5
+                        ]
+                    }
+                }
+            }
+        },
+        "revision": null,
+        "use_imagenet_stats": true,
+        "video_backend": "torchcodec"
+    },
+    "env": null,
+    "policy": {
+        "type": "act",
+        "n_obs_steps": 1,
+        "normalization_mapping": {
+            "VISUAL": "MEAN_STD",
+            "STATE": "MEAN_STD",
+            "ACTION": "MEAN_STD"
+        },
+        "input_features": {
+            "observation.state": {
+                "type": "STATE",
+                "shape": [
+                    7
+                ]
+            },
+            "observation.images.top_camera": {
+                "type": "VISUAL",
+                "shape": [
+                    3,
+                    480,
+                    640
+                ]
+            },
+            "observation.images.bottom_camera": {
+                "type": "VISUAL",
+                "shape": [
+                    3,
+                    480,
+                    640
+                ]
+            },
+            "observation.images.right": {
+                "type": "VISUAL",
+                "shape": [
+                    3,
+                    480,
+                    640
+                ]
+            }
+        },
+        "output_features": {
+            "action": {
+                "type": "ACTION",
+                "shape": [
+                    7
+                ]
+            }
+        },
+        "device": "cuda",
+        "use_amp": false,
+        "chunk_size": 100,
+        "n_action_steps": 100,
+        "vision_backbone": "resnet18",
+        "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+        "replace_final_stride_with_dilation": false,
+        "pre_norm": false,
+        "dim_model": 512,
+        "n_heads": 8,
+        "dim_feedforward": 3200,
+        "feedforward_activation": "relu",
+        "n_encoder_layers": 4,
+        "n_decoder_layers": 1,
+        "use_vae": true,
+        "latent_dim": 32,
+        "n_vae_encoder_layers": 4,
+        "temporal_ensemble_coeff": null,
+        "dropout": 0.1,
+        "kl_weight": 10.0,
+        "optimizer_lr": 1e-05,
+        "optimizer_weight_decay": 0.0001,
+        "optimizer_lr_backbone": 1e-05
+    },
+    "output_dir": "outputs/train/sam_single_cube",
+    "job_name": "sam_single_cube",
+    "resume": false,
+    "seed": 1000,
+    "num_workers": 4,
+    "batch_size": 8,
+    "steps": 600,
+    "eval_freq": 20000,
+    "log_freq": 200,
+    "save_checkpoint": true,
+    "save_freq": 600,
+    "use_policy_training_preset": true,
+    "optimizer": {
+        "type": "adamw",
+        "lr": 1e-05,
+        "weight_decay": 0.0001,
+        "grad_clip_norm": 10.0,
+        "betas": [
+            0.9,
+            0.999
+        ],
+        "eps": 1e-08
+    },
+    "scheduler": null,
+    "eval": {
+        "n_episodes": 50,
+        "batch_size": 50,
+        "use_async_envs": false
+    },
+    "wandb": {
+        "enable": true,
+        "disable_artifact": false,
+        "project": "lerobot",
+        "entity": null,
+        "notes": null,
+        "run_id": "sam_single_cube",
+        "mode": null
+    }
+}

checkpoints/000600/training_state/optimizer_param_groups.json

@@ -0,0 +1,189 @@
+[
+    {
+        "lr": 1e-05,
+        "betas": [
+            0.9,
+            0.999
+        ],
+        "eps": 1e-08,
+        "weight_decay": 0.0001,
+        "amsgrad": false,
+        "foreach": null,
+        "maximize": false,
+        "capturable": false,
+        "differentiable": false,
+        "fused": null,
+        "params": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31,
+            32,
+            33,
+            34,
+            35,
+            36,
+            37,
+            38,
+            39,
+            40,
+            41,
+            42,
+            43,
+            44,
+            45,
+            46,
+            47,
+            48,
+            49,
+            50,
+            51,
+            52,
+            53,
+            54,
+            55,
+            56,
+            57,
+            58,
+            59,
+            60,
+            61,
+            62,
+            63,
+            64,
+            65,
+            66,
+            67,
+            68,
+            69,
+            70,
+            71,
+            72,
+            73,
+            74,
+            75,
+            76,
+            77,
+            78,
+            79,
+            80,
+            81,
+            82,
+            83,
+            84,
+            85,
+            86,
+            87,
+            88,
+            89,
+            90,
+            91,
+            92,
+            93,
+            94,
+            95,
+            96,
+            97,
+            98,
+            99,
+            100,
+            101,
+            102,
+            103,
+            104,
+            105,
+            106,
+            107,
+            108,
+            109,
+            110,
+            111,
+            112,
+            113,
+            114,
+            115,
+            116,
+            117,
+            118,
+            119,
+            120,
+            121,
+            122,
+            123,
+            124,
+            125,
+            126,
+            127,
+            128,
+            129,
+            130,
+            131,
+            132
+        ]
+    },
+    {
+        "lr": 1e-05,
+        "betas": [
+            0.9,
+            0.999
+        ],
+        "eps": 1e-08,
+        "weight_decay": 0.0001,
+        "amsgrad": false,
+        "foreach": null,
+        "maximize": false,
+        "capturable": false,
+        "differentiable": false,
+        "fused": null,
+        "params": [
+            133,
+            134,
+            135,
+            136,
+            137,
+            138,
+            139,
+            140,
+            141,
+            142,
+            143,
+            144,
+            145,
+            146,
+            147,
+            148,
+            149,
+            150,
+            151,
+            152
+        ]
+    }
+]

checkpoints/000600/training_state/optimizer_state.safetensors

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

checkpoints/000600/training_state/rng_state.safetensors

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

checkpoints/000600/training_state/training_step.json

@@ -0,0 +1,3 @@
+{
+    "step": 600
+}

wandb/debug-internal.log

@@ -0,0 +1,15 @@
+{"time":"2025-06-12T23:27:15.538445428Z","level":"INFO","msg":"stream: starting","core version":"0.19.11","symlink path":"outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug-core.log"}
+{"time":"2025-06-12T23:27:15.785939368Z","level":"INFO","msg":"created new stream","id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.785984919Z","level":"INFO","msg":"stream: started","id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786041262Z","level":"INFO","msg":"sender: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786041501Z","level":"INFO","msg":"writer: Do: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786093976Z","level":"INFO","msg":"handler: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:16.010724573Z","level":"INFO","msg":"Starting system monitor"}
+{"time":"2025-06-12T23:28:50.664224333Z","level":"INFO","msg":"stream: closing","id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:50.664258051Z","level":"INFO","msg":"Stopping system monitor"}
+{"time":"2025-06-12T23:28:50.664326487Z","level":"INFO","msg":"Stopped system monitor"}
+{"time":"2025-06-12T23:28:58.708086923Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-06-12T23:28:58.880071508Z","level":"INFO","msg":"handler: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.880113145Z","level":"INFO","msg":"writer: Close: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.880151088Z","level":"INFO","msg":"sender: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.8801821Z","level":"INFO","msg":"stream: closed","id":"sam_single_cube"}

wandb/debug.log

@@ -0,0 +1,23 @@
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Current SDK version is 0.19.11
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Configure stats pid to 2971
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from /root/.config/wandb/settings
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from /content/wandb/settings
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from environment variables
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:setup_run_log_directory():724] Logging user logs to outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug.log
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:setup_run_log_directory():725] Logging internal logs to outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug-internal.log
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:init():852] calling init triggers
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():857] wandb.init called with sweep_config: {}
+config: {'dataset': {'repo_id': 'girardijp/sam_pick_cube_single', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec'}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'chunk_size': 100, 'n_action_steps': 100, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'replace_final_stride_with_dilation': False, 'pre_norm': False, 'dim_model': 512, 'n_heads': 8, 'dim_feedforward': 3200, 'feedforward_activation': 'relu', 'n_encoder_layers': 4, 'n_decoder_layers': 1, 'use_vae': True, 'latent_dim': 32, 'n_vae_encoder_layers': 4, 'temporal_ensemble_coeff': None, 'dropout': 0.1, 'kl_weight': 10.0, 'optimizer_lr': 1e-05, 'optimizer_weight_decay': 0.0001, 'optimizer_lr_backbone': 1e-05}, 'output_dir': 'outputs/train/sam_single_cube', 'job_name': 'sam_single_cube', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 8, 'steps': 600, 'eval_freq': 20000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 600, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 1e-05, 'weight_decay': 0.0001, 'grad_clip_norm': 10.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': None, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': False}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': 'sam_single_cube', 'mode': None}, '_wandb': {}}
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():893] starting backend
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():897] sending inform_init request
+2025-06-12 23:27:15,535 INFO    MainThread:2971 [backend.py:_multiprocessing_setup():101] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2025-06-12 23:27:15,535 INFO    MainThread:2971 [wandb_init.py:init():907] backend started and connected
+2025-06-12 23:27:15,538 INFO    MainThread:2971 [wandb_init.py:init():1005] updated telemetry
+2025-06-12 23:27:15,539 INFO    MainThread:2971 [wandb_init.py:init():1029] communicating run to backend with 90.0 second timeout
+2025-06-12 23:27:16,005 INFO    MainThread:2971 [wandb_init.py:init():1104] starting run threads in backend
+2025-06-12 23:27:16,433 INFO    MainThread:2971 [wandb_run.py:_console_start():2573] atexit reg
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2421] redirect: wrap_raw
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2490] Wrapping output streams.
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2513] Redirects installed.
+2025-06-12 23:27:16,439 INFO    MainThread:2971 [wandb_init.py:init():1150] run started, returning control to user process
+2025-06-12 23:28:50,663 INFO    MsgRouterThr:2971 [mailbox.py:close():129] [no run ID] Closing mailbox, abandoning 2 handles.

wandb/run-20250612_232715-sam_single_cube/files/config.yaml

@@ -0,0 +1,167 @@
+_wandb:
+    value:
+        cli_version: 0.19.11
+        m: []
+        python_version: 3.11.13
+        t:
+            "1":
+                - 1
+                - 41
+                - 49
+                - 51
+                - 55
+                - 105
+            "2":
+                - 1
+                - 41
+                - 49
+                - 51
+                - 55
+                - 105
+            "3":
+                - 13
+                - 14
+                - 15
+                - 16
+                - 23
+                - 55
+                - 61
+            "4": 3.11.13
+            "5": 0.19.11
+            "8":
+                - 5
+            "10":
+                - 21
+            "12": 0.19.11
+            "13": linux-x86_64
+batch_size:
+    value: 8
+dataset:
+    value:
+        episodes: null
+        image_transforms:
+            enable: false
+            max_num_transforms: 3
+            random_order: false
+            tfs:
+                brightness:
+                    kwargs:
+                        brightness:
+                            - 0.8
+                            - 1.2
+                    type: ColorJitter
+                    weight: 1
+                contrast:
+                    kwargs:
+                        contrast:
+                            - 0.8
+                            - 1.2
+                    type: ColorJitter
+                    weight: 1
+                hue:
+                    kwargs:
+                        hue:
+                            - -0.05
+                            - 0.05
+                    type: ColorJitter
+                    weight: 1
+                saturation:
+                    kwargs:
+                        saturation:
+                            - 0.5
+                            - 1.5
+                    type: ColorJitter
+                    weight: 1
+                sharpness:
+                    kwargs:
+                        sharpness:
+                            - 0.5
+                            - 1.5
+                    type: SharpnessJitter
+                    weight: 1
+        repo_id: girardijp/sam_pick_cube_single
+        revision: null
+        root: null
+        use_imagenet_stats: true
+        video_backend: torchcodec
+env:
+    value: null
+eval:
+    value:
+        batch_size: 50
+        n_episodes: 50
+        use_async_envs: false
+eval_freq:
+    value: 20000
+job_name:
+    value: sam_single_cube
+log_freq:
+    value: 200
+num_workers:
+    value: 4
+optimizer:
+    value:
+        betas:
+            - 0.9
+            - 0.999
+        eps: 1e-08
+        grad_clip_norm: 10
+        lr: 1e-05
+        type: adamw
+        weight_decay: 0.0001
+output_dir:
+    value: outputs/train/sam_single_cube
+policy:
+    value:
+        chunk_size: 100
+        device: cuda
+        dim_feedforward: 3200
+        dim_model: 512
+        dropout: 0.1
+        feedforward_activation: relu
+        kl_weight: 10
+        latent_dim: 32
+        n_action_steps: 100
+        n_decoder_layers: 1
+        n_encoder_layers: 4
+        n_heads: 8
+        n_obs_steps: 1
+        n_vae_encoder_layers: 4
+        normalization_mapping:
+            ACTION: MEAN_STD
+            STATE: MEAN_STD
+            VISUAL: MEAN_STD
+        optimizer_lr: 1e-05
+        optimizer_lr_backbone: 1e-05
+        optimizer_weight_decay: 0.0001
+        pre_norm: false
+        pretrained_backbone_weights: ResNet18_Weights.IMAGENET1K_V1
+        replace_final_stride_with_dilation: false
+        temporal_ensemble_coeff: null
+        type: act
+        use_amp: false
+        use_vae: true
+        vision_backbone: resnet18
+resume:
+    value: false
+save_checkpoint:
+    value: true
+save_freq:
+    value: 600
+scheduler:
+    value: null
+seed:
+    value: 1000
+steps:
+    value: 600
+use_policy_training_preset:
+    value: true
+wandb:
+    value:
+        disable_artifact: false
+        enable: true
+        entity: null
+        mode: null
+        notes: null
+        project: lerobot
+        run_id: sam_single_cube

wandb/run-20250612_232715-sam_single_cube/files/output.log

@@ -0,0 +1,246 @@
+Logs will be synced with wandb.
+INFO 2025-06-12 23:27:16 ndb_utils.py:96 Track this run --> https://wandb.ai/girardijp3-universidade-federal-de-santa-catarina/lerobot/runs/sam_single_cube
+INFO 2025-06-12 23:27:16 ts/train.py:127 Creating dataset
+info.json: 100% 3.94k/3.94k [00:00<00:00, 16.5MB/s]
+tasks.jsonl: 100% 46.0/46.0 [00:00<00:00, 161kB/s]
+episodes.jsonl: 100% 3.66k/3.66k [00:00<00:00, 15.3MB/s]
+episodes_stats.jsonl: 100% 134k/134k [00:00<00:00, 1.56MB/s]
+Fetching 4 files: 100% 4/4 [00:00<00:00,  9.93it/s]]
+.gitattributes: 100% 2.46k/2.46k [00:00<00:00, 8.21MB/s]
+README.md: 100% 4.45k/4.45k [00:00<00:00, 12.2MB/s]t/s]
+episode_000000.parquet: 100% 23.8k/23.8k [00:00<00:00, 50.0MB/s]
+episode_000005.parquet: 100% 25.9k/25.9k [00:00<00:00, 56.1MB/s]
+episode_000001.parquet: 100% 26.1k/26.1k [00:00<00:00, 54.7MB/s]
+episode_000004.parquet: 100% 26.1k/26.1k [00:00<00:00, 65.5MB/s]
+episode_000003.parquet: 100% 28.6k/28.6k [00:00<00:00, 68.4MB/s]
+episode_000002.parquet: 100% 27.1k/27.1k [00:00<00:00, 50.7MB/s]
+episode_000007.parquet: 100% 28.8k/28.8k [00:00<00:00, 55.4MB/s]
+episode_000009.parquet: 100% 24.6k/24.6k [00:00<00:00, 60.2MB/s]
+episode_000010.parquet: 100% 24.8k/24.8k [00:00<00:00, 60.6MB/s]
+episode_000008.parquet: 100% 27.6k/27.6k [00:00<00:00, 67.2MB/s]
+episode_000011.parquet: 100% 26.3k/26.3k [00:00<00:00, 71.8MB/s]
+episode_000012.parquet: 100% 26.7k/26.7k [00:00<00:00, 62.3MB/s]
+episode_000013.parquet: 100% 27.6k/27.6k [00:00<00:00, 60.4MB/s]
+episode_000006.parquet: 100% 27.8k/27.8k [00:00<00:00, 44.1MB/s]
+episode_000017.parquet: 100% 28.3k/28.3k [00:00<00:00, 68.2MB/s]
+episode_000014.parquet: 100% 27.3k/27.3k [00:00<00:00, 63.8MB/s]
+episode_000016.parquet: 100% 27.5k/27.5k [00:00<00:00, 66.8MB/s]
+episode_000015.parquet: 100% 26.1k/26.1k [00:00<00:00, 58.8MB/s]
+episode_000019.parquet: 100% 25.4k/25.4k [00:00<00:00, 72.1MB/s]
+episode_000020.parquet: 100% 26.6k/26.6k [00:00<00:00, 60.2MB/s]
+episode_000021.parquet: 100% 25.0k/25.0k [00:00<00:00, 69.7MB/s]
+episode_000018.parquet: 100% 24.4k/24.4k [00:00<00:00, 47.2MB/s]
+episode_000024.parquet: 100% 25.4k/25.4k [00:00<00:00, 56.2MB/s]
+episode_000025.parquet: 100% 25.9k/25.9k [00:00<00:00, 62.7MB/s]
+episode_000022.parquet: 100% 24.8k/24.8k [00:00<00:00, 60.2MB/s]
+episode_000023.parquet: 100% 26.8k/26.8k [00:00<00:00, 66.5MB/s]
+episode_000026.parquet: 100% 26.9k/26.9k [00:00<00:00, 74.3MB/s]
+episode_000027.parquet: 100% 24.9k/24.9k [00:00<00:00, 66.2MB/s]
+episode_000029.parquet: 100% 25.2k/25.2k [00:00<00:00, 49.9MB/s]
+episode_000031.parquet: 100% 26.0k/26.0k [00:00<00:00, 14.2MB/s]
+episode_000030.parquet: 100% 24.5k/24.5k [00:00<00:00, 22.7MB/s]
+episode_000033.parquet: 100% 24.1k/24.1k [00:00<00:00, 7.64MB/s]
+episode_000032.parquet: 100% 26.1k/26.1k [00:00<00:00, 15.7MB/s]
+episode_000028.parquet: 100% 24.8k/24.8k [00:00<00:00, 57.1MB/s]
+episode_000034.parquet: 100% 25.8k/25.8k [00:00<00:00, 60.6MB/s]
+episode_000035.parquet: 100% 25.5k/25.5k [00:00<00:00, 71.3MB/s]
+episode_000036.parquet: 100% 26.7k/26.7k [00:00<00:00, 58.8MB/s]
+episode_000037.parquet: 100% 24.5k/24.5k [00:00<00:00, 69.9MB/s]
+episode_000042.parquet: 100% 24.3k/24.3k [00:00<00:00, 64.4MB/s]
+episode_000038.parquet: 100% 26.3k/26.3k [00:00<00:00, 65.4MB/s]
+episode_000040.parquet: 100% 25.2k/25.2k [00:00<00:00, 69.7MB/s]
+episode_000039.parquet: 100% 26.9k/26.9k [00:00<00:00, 70.3MB/s]
+episode_000041.parquet: 100% 24.6k/24.6k [00:00<00:00, 70.8MB/s]
+episode_000043.parquet: 100% 25.1k/25.1k [00:00<00:00, 66.9MB/s]
+episode_000044.parquet: 100% 25.1k/25.1k [00:00<00:00, 66.8MB/s]
+episode_000045.parquet: 100% 23.7k/23.7k [00:00<00:00, 60.5MB/s]
+episode_000046.parquet: 100% 23.4k/23.4k [00:00<00:00, 66.7MB/s]
+episode_000049.parquet: 100% 24.1k/24.1k [00:00<00:00, 73.9MB/s]
+episode_000050.parquet: 100% 25.3k/25.3k [00:00<00:00, 62.6MB/s]
+episode_000048.parquet: 100% 24.7k/24.7k [00:00<00:00, 61.0MB/s]
+episode_000052.parquet: 100% 24.2k/24.2k [00:00<00:00, 64.1MB/s]
+episode_000047.parquet: 100% 25.1k/25.1k [00:00<00:00, 70.9MB/s]
+episode_000051.parquet: 100% 25.4k/25.4k [00:00<00:00, 42.5MB/s]
+episode_000053.parquet: 100% 25.5k/25.5k [00:00<00:00, 66.3MB/s]
+episode_000000.mp4: 100% 3.52M/3.52M [00:00<00:00, 21.6MB/s]
+episode_000007.mp4: 100% 3.63M/3.63M [00:00<00:00, 38.6MB/s]
+episode_000004.mp4: 100% 3.57M/3.57M [00:00<00:00, 17.9MB/s]
+episode_000003.mp4: 100% 3.58M/3.58M [00:00<00:00, 39.5MB/s]
+episode_000002.mp4: 100% 3.52M/3.52M [00:00<00:00, 16.7MB/s]
+episode_000005.mp4: 100% 3.57M/3.57M [00:00<00:00, 12.7MB/s]
+episode_000001.mp4: 100% 3.52M/3.52M [00:00<00:00, 10.0MB/s]
+episode_000008.mp4: 100% 3.64M/3.64M [00:00<00:00, 17.9MB/s]
+episode_000006.mp4: 100% 3.60M/3.60M [00:00<00:00, 6.59MB/s]
+episode_000009.mp4: 100% 3.62M/3.62M [00:00<00:00, 38.6MB/s]
+episode_000011.mp4: 100% 3.61M/3.61M [00:00<00:00, 19.2MB/s]
+episode_000012.mp4: 100% 3.67M/3.67M [00:00<00:00, 26.5MB/s]
+episode_000010.mp4: 100% 3.61M/3.61M [00:00<00:00, 15.8MB/s]
+episode_000013.mp4: 100% 3.65M/3.65M [00:00<00:00, 15.4MB/s]
+episode_000014.mp4: 100% 3.65M/3.65M [00:00<00:00, 39.5MB/s]
+episode_000015.mp4: 100% 3.63M/3.63M [00:00<00:00, 22.2MB/s]
+episode_000018.mp4: 100% 3.59M/3.59M [00:00<00:00, 104MB/s]]
+episode_000019.mp4: 100% 3.63M/3.63M [00:00<00:00, 64.7MB/s]
+episode_000016.mp4: 100% 3.65M/3.65M [00:00<00:00, 13.1MB/s]
+episode_000017.mp4: 100% 3.67M/3.67M [00:00<00:00, 12.7MB/s]
+episode_000021.mp4: 100% 3.64M/3.64M [00:00<00:00, 18.9MB/s]
+episode_000020.mp4: 100% 3.61M/3.61M [00:00<00:00, 19.3MB/s]
+episode_000022.mp4: 100% 3.60M/3.60M [00:00<00:00, 20.4MB/s]
+episode_000023.mp4: 100% 3.63M/3.63M [00:00<00:00, 69.8MB/s]
+episode_000024.mp4: 100% 3.61M/3.61M [00:00<00:00, 36.4MB/s]
+episode_000025.mp4: 100% 3.61M/3.61M [00:00<00:00, 33.1MB/s]
+episode_000026.mp4: 100% 3.63M/3.63M [00:00<00:00, 19.3MB/s]
+episode_000027.mp4: 100% 3.63M/3.63M [00:00<00:00, 25.6MB/s]
+episode_000029.mp4: 100% 3.59M/3.59M [00:00<00:00, 21.7MB/s]
+episode_000028.mp4: 100% 3.60M/3.60M [00:00<00:00, 31.5MB/s]
+episode_000031.mp4: 100% 3.66M/3.66M [00:00<00:00, 26.8MB/s]
+episode_000030.mp4: 100% 3.64M/3.64M [00:00<00:00, 18.5MB/s]
+episode_000033.mp4: 100% 3.63M/3.63M [00:00<00:00, 23.4MB/s]
+episode_000032.mp4: 100% 3.63M/3.63M [00:00<00:00, 13.1MB/s]
+episode_000034.mp4: 100% 3.64M/3.64M [00:00<00:00, 19.5MB/s]
+episode_000036.mp4: 100% 3.65M/3.65M [00:00<00:00, 31.7MB/s]
+episode_000037.mp4: 100% 3.63M/3.63M [00:00<00:00, 25.3MB/s]
+episode_000039.mp4: 100% 3.67M/3.67M [00:00<00:00, 32.4MB/s]
+episode_000035.mp4: 100% 3.63M/3.63M [00:00<00:00, 7.71MB/s]
+episode_000038.mp4: 100% 3.60M/3.60M [00:00<00:00, 13.0MB/s]
+episode_000040.mp4: 100% 3.61M/3.61M [00:00<00:00, 31.9MB/s]
+episode_000041.mp4: 100% 3.64M/3.64M [00:00<00:00, 53.0MB/s]
+episode_000043.mp4: 100% 3.65M/3.65M [00:00<00:00, 38.6MB/s]
+episode_000042.mp4: 100% 3.62M/3.62M [00:00<00:00, 33.1MB/s]
+episode_000044.mp4: 100% 3.66M/3.66M [00:00<00:00, 16.7MB/s]
+episode_000045.mp4: 100% 3.63M/3.63M [00:00<00:00, 18.4MB/s]
+episode_000047.mp4: 100% 3.65M/3.65M [00:00<00:00, 38.3MB/s]
+episode_000048.mp4: 100% 3.65M/3.65M [00:00<00:00, 19.0MB/s]
+episode_000049.mp4: 100% 3.65M/3.65M [00:00<00:00, 18.9MB/s]
+episode_000050.mp4: 100% 3.63M/3.63M [00:00<00:00, 28.8MB/s]
+episode_000046.mp4: 100% 3.60M/3.60M [00:00<00:00, 6.53MB/s]
+episode_000000.mp4: 100% 1.88M/1.88M [00:00<00:00, 131MB/s]]
+episode_000053.mp4: 100% 3.66M/3.66M [00:00<00:00, 36.6MB/s]
+episode_000001.mp4: 100% 1.84M/1.84M [00:00<00:00, 15.7MB/s]
+episode_000051.mp4: 100% 3.65M/3.65M [00:00<00:00, 7.83MB/s]
+episode_000052.mp4: 100% 3.64M/3.64M [00:00<00:00, 19.6MB/s]
+episode_000002.mp4: 100% 1.73M/1.73M [00:00<00:00, 9.50MB/s]
+episode_000003.mp4: 100% 1.70M/1.70M [00:00<00:00, 111MB/s]
+episode_000004.mp4: 100% 1.70M/1.70M [00:00<00:00, 18.2MB/s]
+episode_000005.mp4: 100% 1.73M/1.73M [00:00<00:00, 18.8MB/s]
+episode_000007.mp4: 100% 1.66M/1.66M [00:00<00:00, 112MB/s]]
+episode_000006.mp4: 100% 1.74M/1.74M [00:00<00:00, 65.9MB/s]
+episode_000008.mp4: 100% 1.68M/1.68M [00:00<00:00, 53.8MB/s]
+episode_000009.mp4: 100% 1.84M/1.84M [00:00<00:00, 71.2MB/s]
+episode_000012.mp4: 100% 1.69M/1.69M [00:00<00:00, 113MB/s]
+episode_000013.mp4: 100% 1.73M/1.73M [00:00<00:00, 38.1MB/s]
+episode_000010.mp4: 100% 1.89M/1.89M [00:00<00:00, 5.17MB/s]
+episode_000014.mp4: 100% 1.69M/1.69M [00:00<00:00, 17.3MB/s]
+episode_000016.mp4: 100% 1.71M/1.71M [00:00<00:00, 65.6MB/s]
+episode_000011.mp4: 100% 1.91M/1.91M [00:00<00:00, 5.27MB/s]
+episode_000015.mp4: 100% 1.71M/1.71M [00:00<00:00, 18.4MB/s]
+episode_000018.mp4: 100% 1.83M/1.83M [00:00<00:00, 31.4MB/s]
+episode_000017.mp4: 100% 1.69M/1.69M [00:00<00:00, 14.4MB/s]
+episode_000021.mp4: 100% 1.74M/1.74M [00:00<00:00, 87.7MB/s]
+episode_000023.mp4: 100% 1.72M/1.72M [00:00<00:00, 81.6MB/s]
+episode_000024.mp4: 100% 1.81M/1.81M [00:00<00:00, 17.1MB/s]
+episode_000025.mp4: 100% 1.75M/1.75M [00:00<00:00, 102MB/s]
+episode_000026.mp4: 100% 1.68M/1.68M [00:00<00:00, 67.0MB/s]
+episode_000020.mp4: 100% 1.83M/1.83M [00:00<00:00, 6.51MB/s]
+episode_000022.mp4: 100% 1.79M/1.79M [00:00<00:00, 4.74MB/s]
+episode_000027.mp4: 100% 1.79M/1.79M [00:00<00:00, 66.5MB/s]
+episode_000019.mp4: 100% 1.79M/1.79M [00:00<00:00, 4.77MB/s]
+episode_000031.mp4: 100% 1.76M/1.76M [00:00<00:00, 100MB/s]]
+episode_000028.mp4: 100% 1.89M/1.89M [00:00<00:00, 19.0MB/s]
+episode_000030.mp4: 100% 1.75M/1.75M [00:00<00:00, 17.4MB/s]
+episode_000032.mp4: 100% 1.79M/1.79M [00:00<00:00, 49.6MB/s]
+episode_000034.mp4: 100% 1.75M/1.75M [00:00<00:00, 49.5MB/s]
+episode_000033.mp4: 100% 1.76M/1.76M [00:00<00:00, 47.7MB/s]
+episode_000036.mp4: 100% 1.68M/1.68M [00:00<00:00, 148MB/s]
+episode_000029.mp4: 100% 1.84M/1.84M [00:00<00:00, 4.97MB/s]
+episode_000038.mp4: 100% 1.89M/1.89M [00:00<00:00, 54.3MB/s]
+episode_000037.mp4: 100% 1.85M/1.85M [00:00<00:00, 19.8MB/s]
+episode_000035.mp4: 100% 1.82M/1.82M [00:00<00:00, 41.1MB/s]
+episode_000039.mp4: 100% 1.77M/1.77M [00:00<00:00, 129MB/s]
+episode_000040.mp4: 100% 1.82M/1.82M [00:00<00:00, 69.8MB/s]
+episode_000041.mp4: 100% 1.81M/1.81M [00:00<00:00, 64.4MB/s]
+episode_000042.mp4: 100% 1.92M/1.92M [00:00<00:00, 19.4MB/s]
+episode_000044.mp4: 100% 1.74M/1.74M [00:00<00:00, 88.0MB/s]
+episode_000046.mp4: 100% 1.78M/1.78M [00:00<00:00, 121MB/s]
+episode_000045.mp4: 100% 1.73M/1.73M [00:00<00:00, 60.0MB/s]
+episode_000043.mp4: 100% 1.79M/1.79M [00:00<00:00, 9.87MB/s]
+episode_000047.mp4: 100% 1.75M/1.75M [00:00<00:00, 19.0MB/s]
+episode_000050.mp4: 100% 1.75M/1.75M [00:00<00:00, 97.2MB/s]
+episode_000048.mp4: 100% 1.72M/1.72M [00:00<00:00, 17.6MB/s]
+episode_000051.mp4: 100% 1.72M/1.72M [00:00<00:00, 53.9MB/s]
+episode_000049.mp4: 100% 1.70M/1.70M [00:00<00:00, 18.3MB/s]
+episode_000053.mp4: 100% 1.80M/1.80M [00:00<00:00, 54.3MB/s]
+episode_000052.mp4: 100% 1.91M/1.91M [00:00<00:00, 20.6MB/s]
+episode_000001.mp4: 100% 2.51M/2.51M [00:00<00:00, 36.9MB/s]
+episode_000000.mp4: 100% 2.47M/2.47M [00:00<00:00, 25.1MB/s]
+episode_000002.mp4: 100% 2.58M/2.58M [00:00<00:00, 17.3MB/s]
+episode_000004.mp4: 100% 2.57M/2.57M [00:00<00:00, 24.4MB/s]
+episode_000005.mp4: 100% 2.57M/2.57M [00:00<00:00, 23.4MB/s]
+episode_000003.mp4: 100% 2.56M/2.56M [00:00<00:00, 16.3MB/s]
+episode_000008.mp4: 100% 2.59M/2.59M [00:00<00:00, 37.6MB/s]
+episode_000006.mp4: 100% 2.58M/2.58M [00:00<00:00, 13.9MB/s]
+episode_000009.mp4: 100% 2.50M/2.50M [00:00<00:00, 23.4MB/s]
+episode_000011.mp4: 100% 2.55M/2.55M [00:00<00:00, 31.9MB/s]
+episode_000012.mp4: 100% 2.57M/2.57M [00:00<00:00, 38.7MB/s]
+episode_000007.mp4: 100% 2.57M/2.57M [00:00<00:00, 6.82MB/s]
+episode_000010.mp4: 100% 2.54M/2.54M [00:00<00:00, 17.2MB/s]
+episode_000013.mp4: 100% 2.55M/2.55M [00:00<00:00, 22.9MB/s]
+episode_000016.mp4: 100% 2.57M/2.57M [00:00<00:00, 20.6MB/s]
+episode_000015.mp4: 100% 2.55M/2.55M [00:00<00:00, 13.8MB/s]
+episode_000014.mp4: 100% 2.61M/2.61M [00:00<00:00, 25.4MB/s]
+episode_000017.mp4: 100% 2.61M/2.61M [00:00<00:00, 33.5MB/s]
+episode_000018.mp4: 100% 2.48M/2.48M [00:00<00:00, 18.6MB/s]
+episode_000020.mp4: 100% 2.54M/2.54M [00:00<00:00, 15.6MB/s]
+episode_000019.mp4: 100% 2.53M/2.53M [00:00<00:00, 8.36MB/s]
+episode_000022.mp4: 100% 2.51M/2.51M [00:00<00:00, 25.5MB/s]
+episode_000021.mp4: 100% 2.51M/2.51M [00:00<00:00, 12.0MB/s]
+episode_000023.mp4: 100% 2.55M/2.55M [00:00<00:00, 13.3MB/s]
+episode_000024.mp4: 100% 2.52M/2.52M [00:00<00:00, 21.2MB/s]
+episode_000025.mp4: 100% 2.54M/2.54M [00:00<00:00, 13.8MB/s]
+episode_000026.mp4: 100% 2.59M/2.59M [00:00<00:00, 26.3MB/s]
+episode_000028.mp4: 100% 2.52M/2.52M [00:00<00:00, 22.5MB/s]
+episode_000027.mp4: 100% 2.51M/2.51M [00:00<00:00, 14.2MB/s]
+episode_000030.mp4: 100% 2.51M/2.51M [00:00<00:00, 25.4MB/s]
+episode_000031.mp4: 100% 2.53M/2.53M [00:00<00:00, 13.6MB/s]
+episode_000032.mp4: 100% 2.55M/2.55M [00:00<00:00, 27.3MB/s]
+episode_000029.mp4: 100% 2.55M/2.55M [00:00<00:00, 21.8MB/s]
+episode_000033.mp4: 100% 2.53M/2.53M [00:00<00:00, 24.4MB/s]
+episode_000034.mp4: 100% 2.56M/2.56M [00:00<00:00, 20.4MB/s]
+episode_000036.mp4: 100% 2.55M/2.55M [00:00<00:00, 25.7MB/s]
+episode_000035.mp4: 100% 2.54M/2.54M [00:00<00:00, 13.7MB/s]
+episode_000037.mp4: 100% 2.52M/2.52M [00:00<00:00, 22.0MB/s]
+episode_000038.mp4: 100% 2.54M/2.54M [00:00<00:00, 25.5MB/s]
+episode_000039.mp4: 100% 2.54M/2.54M [00:00<00:00, 10.9MB/s]
+episode_000040.mp4: 100% 2.53M/2.53M [00:00<00:00, 9.00MB/s]
+episode_000041.mp4: 100% 2.53M/2.53M [00:00<00:00, 11.3MB/s]
+episode_000044.mp4: 100% 2.52M/2.52M [00:00<00:00, 27.2MB/s]
+episode_000043.mp4: 100% 2.49M/2.49M [00:00<00:00, 20.5MB/s]
+episode_000042.mp4: 100% 2.49M/2.49M [00:00<00:00, 18.3MB/s]
+episode_000045.mp4: 100% 2.52M/2.52M [00:00<00:00, 15.6MB/s]
+episode_000046.mp4: 100% 2.52M/2.52M [00:00<00:00, 12.1MB/s]
+episode_000050.mp4: 100% 2.52M/2.52M [00:00<00:00, 63.8MB/s]
+episode_000047.mp4: 100% 2.57M/2.57M [00:00<00:00, 23.1MB/s]
+episode_000052.mp4: 100% 2.50M/2.50M [00:00<00:00, 19.7MB/s]
+episode_000048.mp4: 100% 2.53M/2.53M [00:00<00:00, 11.4MB/s]
+episode_000053.mp4: 100% 2.56M/2.56M [00:00<00:00, 26.5MB/s]
+episode_000049.mp4: 100% 2.55M/2.55M [00:00<00:00, 14.4MB/s]
+episode_000051.mp4: 100% 2.54M/2.54M [00:00<00:00, 8.87MB/s]
+Fetching 222 files: 100% 222/222 [00:13<00:00, 16.58it/s]
+Resolving data files: 100% 54/54 [00:00<00:00, 161089.91it/s]
+Downloading data: 100% 54/54 [00:00<00:00, 193088.16files/s]
+Generating train split: 27325 examples [00:00, 253372.74 examples/s]
+INFO 2025-06-12 23:27:32 ts/train.py:138 Creating policy
+Downloading: "https://download.pytorch.org/models/resnet18-f37072fd.pth" to /root/.cache/torch/hub/checkpoints/resnet18-f37072fd.pth
+100% 44.7M/44.7M [00:00<00:00, 218MB/s]
+INFO 2025-06-12 23:27:33 ts/train.py:144 Creating optimizer and scheduler
+INFO 2025-06-12 23:27:33 ts/train.py:156 Output dir: outputs/train/sam_single_cube
+INFO 2025-06-12 23:27:33 ts/train.py:159 cfg.steps=600 (600)
+INFO 2025-06-12 23:27:33 ts/train.py:160 dataset.num_frames=27325 (27K)
+INFO 2025-06-12 23:27:33 ts/train.py:161 dataset.num_episodes=54
+INFO 2025-06-12 23:27:33 ts/train.py:162 num_learnable_params=51599239 (52M)
+INFO 2025-06-12 23:27:33 ts/train.py:163 num_total_params=51599299 (52M)
+INFO 2025-06-12 23:27:33 ts/train.py:202 Start offline training on a fixed dataset
+INFO 2025-06-12 23:28:00 ts/train.py:232 step:200 smpl:2K ep:3 epch:0.06 loss:6.901 grdn:157.475 lr:1.0e-05 updt_s:0.128 data_s:0.008
+INFO 2025-06-12 23:28:24 ts/train.py:232 step:400 smpl:3K ep:6 epch:0.12 loss:2.958 grdn:86.789 lr:1.0e-05 updt_s:0.118 data_s:0.000
+INFO 2025-06-12 23:28:48 ts/train.py:232 step:600 smpl:5K ep:9 epch:0.18 loss:2.474 grdn:75.187 lr:1.0e-05 updt_s:0.116 data_s:0.000
+INFO 2025-06-12 23:28:48 ts/train.py:241 Checkpoint policy after step 600
+INFO 2025-06-12 23:28:50 ts/train.py:283 End of training

wandb/run-20250612_232715-sam_single_cube/files/requirements.txt

@@ -0,0 +1,698 @@
+typing-inspect==0.9.0
+nvidia-cublas-cu12==12.4.5.8
+nvidia-nvjitlink-cu12==12.4.127
+ffmpeg-python==0.2.0
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+datasets==3.6.0
+deepdiff==8.5.0
+lerobot==0.1.0
+numcodecs==0.16.1
+rerun-sdk==0.23.3
+nvidia-cufft-cu12==11.2.1.3
+mergedeep==1.3.4
+orderly-set==5.4.1
+evdev==1.9.2
+donfig==0.8.1.post1
+fsspec==2025.3.0
+TorchCodec==0.2.0
+nvidia-cuda-cupti-cu12==12.4.127
+inquirerpy==0.3.4
+pyserial==3.5
+pynput==1.8.1
+python-xlib==0.33
+crc32c==2.7.1
+pfzy==0.3.4
+feetech-servo-sdk==1.0.0
+pymunk==6.11.1
+jsonlines==4.0.0
+gymnasium==0.29.1
+zarr==3.0.8
+mypy_extensions==1.1.0
+av==14.4.0
+nvidia-cuda-runtime-cu12==12.4.127
+pyyaml-include==1.4.1
+draccus==0.10.0
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusparse-cu12==12.3.1.170
+pyzmq==26.4.0
+opencv-python==4.11.0.86
+google-colab==1.0.0
+widgetsnbextension==3.6.10
+ipykernel==6.17.1
+certifi==2025.4.26
+pyOpenSSL==24.2.1
+wordcloud==1.9.4
+timm==1.0.15
+tabulate==0.9.0
+inflect==7.5.0
+spacy-loggers==1.0.5
+eerepr==0.1.2
+fastrlock==0.8.3
+einops==0.8.1
+pygit2==1.18.0
+jax-cuda12-plugin==0.5.1
+dask-expr==1.1.21
+rich==13.9.4
+hyperframe==6.1.0
+more-itertools==10.7.0
+future==1.0.0
+geocoder==1.38.1
+ydf==0.12.0
+ptyprocess==0.7.0
+pyviz_comms==3.0.4
+llvmlite==0.43.0
+tf_keras==2.18.0
+tenacity==9.1.2
+thinc==8.3.6
+pycryptodomex==3.23.0
+tiktoken==0.9.0
+blobfile==3.0.0
+prophet==1.1.7
+cryptography==43.0.3
+yfinance==0.2.61
+semantic-version==2.10.0
+xarray==2025.3.1
+requests==2.32.3
+oauthlib==3.2.2
+umap-learn==0.5.7
+keras-hub==0.18.1
+wandb==0.19.11
+earthengine-api==1.5.18
+aiohappyeyeballs==2.6.1
+partd==1.4.2
+etils==1.12.2
+libpysal==4.13.0
+cufflinks==0.17.3
+gradio_client==1.10.1
+highspy==1.10.0
+intel-openmp==2025.1.1
+jiter==0.10.0
+langsmith==0.3.44
+docker-pycreds==0.4.0
+pytz==2025.2
+fastdownload==0.0.7
+optree==0.16.0
+uc-micro-py==1.0.3
+cuda-python==12.6.2.post1
+langcodes==3.5.0
+ipyparallel==8.8.0
+nbformat==5.10.4
+lightgbm==4.5.0
+blinker==1.9.0
+blis==1.3.0
+sphinxcontrib-serializinghtml==2.0.0
+jupytext==1.17.2
+keyrings.google-artifactregistry-auth==1.1.2
+soupsieve==2.7
+geopandas==1.0.1
+albucore==0.0.24
+plotly==5.24.1
+requests-oauthlib==2.0.0
+omegaconf==2.3.0
+pluggy==1.6.0
+gym==0.25.2
+pandas-datareader==0.10.0
+pylibraft-cu12==25.2.0
+betterproto==2.0.0b6
+python-dateutil==2.9.0.post0
+pyerfa==2.0.1.5
+jaraco.functools==4.1.0
+jupyter-client==6.1.12
+linkify-it-py==2.0.3
+cymem==2.0.11
+matplotlib==3.10.0
+python-slugify==8.0.4
+tinycss2==1.4.0
+sentry-sdk==2.29.1
+xgboost==2.1.4
+marisa-trie==1.2.1
+frozendict==2.4.6
+pylibcugraph-cu12==25.2.0
+google-auth-httplib2==0.2.0
+mlxtend==0.23.4
+ply==3.11
+miniKanren==1.0.3
+Pygments==2.19.1
+nbclassic==1.3.1
+gin-config==0.5.0
+GDAL==3.8.4
+matplotlib-inline==0.1.7
+PyDrive2==1.21.3
+pydantic==2.11.5
+text-unidecode==1.3
+confection==0.1.5
+httpcore==1.0.9
+scikit-image==0.25.2
+stringzilla==3.12.5
+starlette==0.46.2
+cloudpickle==3.1.1
+google-auth==2.38.0
+zstandard==0.23.0
+pydotplus==2.0.2
+itsdangerous==2.2.0
+scooby==0.10.1
+nvidia-cusparselt-cu12==0.6.2
+libucxx-cu12==0.42.0
+xyzservices==2025.4.0
+ipython-sql==0.5.0
+GitPython==3.1.44
+simple-parsing==0.1.7
+tensorflow-text==2.18.1
+argon2-cffi==25.1.0
+MarkupSafe==3.0.2
+scikit-learn==1.6.1
+sentence-transformers==4.1.0
+textblob==0.19.0
+nvidia-nccl-cu12==2.21.5
+safehttpx==0.1.6
+pytest==8.3.5
+language_data==1.3.0
+jupyterlab_pygments==0.3.0
+flax==0.10.6
+pygame==2.6.1
+tbb==2022.1.0
+h11==0.16.0
+python-multipart==0.0.20
+cyipopt==1.5.0
+chex==0.1.89
+param==2.2.0
+google-cloud-aiplatform==1.95.1
+numpy==2.0.2
+holidays==0.74
+tensorflow_decision_forests==1.11.0
+peewee==3.18.1
+google-cloud-iam==2.19.0
+astropy-iers-data==0.2025.6.2.0.38.23
+packaging==24.2
+gast==0.6.0
+datascience==0.17.6
+babel==2.17.0
+autograd==1.8.0
+six==1.17.0
+raft-dask-cu12==25.2.0
+tqdm==4.67.1
+optax==0.2.4
+Markdown==3.8
+google-cloud-dataproc==5.18.1
+treelite==4.4.1
+keyring==25.6.0
+sphinxcontrib-qthelp==2.0.0
+click==8.2.1
+pandas==2.2.2
+fastai==2.7.19
+murmurhash==1.0.13
+requests-toolbelt==1.0.0
+db-dtypes==1.4.3
+Sphinx==8.2.3
+google-crc32c==1.7.1
+hpack==4.1.0
+uvicorn==0.34.3
+pandocfilters==1.5.1
+libcuvs-cu12==25.2.1
+googledrivedownloader==1.1.0
+libkvikio-cu12==25.2.1
+tomlkit==0.13.2
+immutabledict==4.2.1
+jupyter-leaflet==0.19.2
+tensorstore==0.1.74
+google-auth-oauthlib==1.2.2
+astunparse==1.6.3
+typeguard==4.4.2
+holoviews==1.20.2
+google-cloud-datastore==2.21.0
+httplib2==0.22.0
+progressbar2==4.5.0
+tornado==6.4.2
+psutil==5.9.5
+parso==0.8.4
+rsa==4.9.1
+Cython==3.0.12
+terminado==0.18.1
+pycparser==2.22
+sentencepiece==0.2.0
+dataproc-spark-connect==0.7.5
+absl-py==1.4.0
+smmap==5.0.2
+google-cloud-bigquery-storage==2.32.0
+cons==0.4.6
+chardet==5.2.0
+seaborn==0.13.2
+groovy==0.1.2
+en_core_web_sm==3.8.0
+cffi==1.17.1
+stanio==0.5.1
+ipython-genutils==0.2.0
+jaraco.context==6.0.1
+dm-tree==0.1.9
+cuml-cu12==25.2.1
+xxhash==3.5.0
+pandas-gbq==0.29.0
+imbalanced-learn==0.13.0
+google-pasta==0.2.0
+rpds-py==0.25.1
+namex==0.1.0
+annotated-types==0.7.0
+notebook==6.5.7
+parsy==2.1
+scs==3.2.7.post2
+tf-slim==1.1.0
+zipp==3.22.0
+simplejson==3.20.1
+tensorflow-probability==0.25.0
+pyparsing==3.2.3
+PyDrive==1.3.1
+jax-cuda12-pjrt==0.5.1
+tcmlib==1.3.0
+importlib_resources==6.5.2
+libcugraph-cu12==25.2.0
+peft==0.15.2
+ipyleaflet==0.19.2
+typer==0.16.0
+shap==0.47.2
+Farama-Notifications==0.0.4
+catalogue==2.0.10
+albumentations==2.0.8
+panel==1.7.1
+moviepy==1.0.3
+alabaster==1.0.0
+multitasking==0.0.11
+sklearn-compat==0.1.3
+cloudpathlib==0.21.1
+ale-py==0.11.1
+ipyfilechooser==0.6.0
+langchain-core==0.3.63
+wheel==0.45.1
+google-cloud-bigquery==3.34.0
+multiprocess==0.70.15
+backcall==0.2.0
+tensorboard==2.18.0
+jupyter_kernel_gateway==2.5.2
+statsmodels==0.14.4
+websockets==15.0.1
+hdbscan==0.8.40
+ucx-py-cu12==0.42.0
+google-resumable-media==2.7.2
+typing_extensions==4.14.0
+notebook_shim==0.2.4
+missingno==0.5.2
+distributed==2024.12.1
+torch==2.6.0+cu124
+logical-unification==0.4.6
+tzdata==2025.2
+pyogrio==0.11.0
+mizani==0.13.5
+sphinxcontrib-devhelp==2.0.0
+etuples==0.3.9
+pydot==3.0.4
+xlrd==2.0.1
+pynvjitlink-cu12==0.6.0
+Werkzeug==3.1.3
+aiofiles==24.1.0
+sphinxcontrib-jsmath==1.0.1
+nvidia-nvtx-cu12==12.4.127
+nest-asyncio==1.6.0
+safetensors==0.5.3
+pymystem3==0.2.0
+pillow==11.2.1
+array_record==0.7.2
+cmdstanpy==1.2.5
+webcolors==24.11.1
+filelock==3.18.0
+fonttools==4.58.1
+platformdirs==4.3.8
+h5py==3.13.0
+duckdb==1.2.2
+PyJWT==2.10.1
+h5netcdf==1.6.1
+openpyxl==3.1.5
+protobuf==5.29.5
+geopy==2.4.1
+nvidia-ml-py==12.575.51
+sklearn-pandas==2.2.0
+tensorflow-hub==0.16.1
+gspread-dataframe==4.0.0
+ipywidgets==7.7.1
+libraft-cu12==25.2.0
+fastapi==0.115.12
+ml-dtypes==0.4.1
+yellowbrick==1.5
+locket==1.0.0
+tables==3.10.2
+hf_transfer==0.1.9
+sqlglot==25.20.2
+pytensor==2.31.3
+httpx==0.28.1
+imagesize==1.4.1
+PyWavelets==1.8.0
+geemap==0.35.3
+cudf-polars-cu12==25.2.2
+tokenizers==0.21.1
+google-api-core==2.25.0
+pyasn1_modules==0.4.2
+jsonschema==4.24.0
+graphviz==0.20.3
+websocket-client==1.8.0
+prettytable==3.16.0
+tifffile==2025.6.1
+hyperopt==0.2.7
+toml==0.10.2
+tzlocal==5.3.1
+imageio==2.37.0
+traittypes==0.2.1
+pyproj==3.7.1
+simsimd==6.4.7
+tensorflow-datasets==4.9.9
+soundfile==0.13.1
+clarabel==0.11.0
+uritemplate==4.2.0
+lxml==5.4.0
+shellingham==1.5.4
+osqp==1.0.4
+numexpr==2.10.2
+ratelim==0.1.6
+mdurl==0.1.2
+flatbuffers==25.2.10
+regex==2024.11.6
+gitdb==4.0.12
+numba-cuda==0.2.0
+sphinxcontrib-applehelp==2.0.0
+keras-nlp==0.18.1
+spacy-legacy==3.0.12
+geographiclib==2.0
+orjson==3.10.18
+mdit-py-plugins==0.4.2
+triton==3.2.0
+colorcet==3.1.0
+jaxlib==0.5.1
+psycopg2==2.9.10
+googleapis-common-protos==1.70.0
+nx-cugraph-cu12==25.2.0
+jsonpointer==3.0.0
+proglog==0.1.12
+kiwisolver==1.4.8
+python-louvain==0.16
+tblib==3.1.0
+google-cloud-storage==2.19.0
+diffusers==0.33.1
+google-api-python-client==2.171.0
+libclang==18.1.1
+altair==5.5.0
+libcudf-cu12==25.2.1
+CacheControl==0.14.3
+editdistance==0.8.1
+orbax-checkpoint==0.11.13
+joblib==1.5.1
+libucx-cu12==1.18.1
+pyasn1==0.6.1
+librosa==0.11.0
+pymc==5.23.0
+tensorboard-data-server==0.7.2
+nltk==3.9.1
+google-cloud-core==2.4.3
+roman-numerals-py==3.1.0
+promise==2.3
+patsy==1.0.1
+typing-inspection==0.4.1
+oauth2client==4.1.3
+distro==1.9.0
+proto-plus==1.26.1
+yarl==1.20.0
+nvtx==0.2.12
+gcsfs==2025.3.2
+google-cloud-functions==1.20.3
+setproctitle==1.3.6
+argon2-cffi-bindings==21.2.0
+pycairo==1.28.0
+accelerate==1.7.0
+dill==0.3.7
+rapids-dask-dependency==25.2.0
+entrypoints==0.4
+google-ai-generativelanguage==0.6.15
+arviz==0.21.0
+grpc-google-iam-v1==0.14.2
+nvidia-cuda-nvcc-cu12==12.5.82
+PyYAML==6.0.2
+torchao==0.10.0
+google-cloud-translate==3.20.2
+curl_cffi==0.11.1
+astropy==7.1.0
+transformers==4.52.4
+audioread==3.0.1
+charset-normalizer==3.4.2
+urllib3==2.4.0
+torchaudio==2.6.0+cu124
+gradio==5.31.0
+torchvision==0.21.0+cu124
+jsonpickle==4.1.1
+bqplot==0.12.45
+pyarrow==18.1.0
+ruff==0.11.12
+music21==9.3.0
+wasabi==1.1.3
+rpy2==3.5.17
+python-snappy==0.7.3
+Jinja2==3.1.6
+pydata-google-auth==1.9.1
+networkx==3.5
+pandas-stubs==2.2.2.240909
+openai==1.84.0
+aiohttp==3.11.15
+pynndescent==0.5.13
+fastprogress==1.0.3
+h2==4.2.0
+intel-cmplr-lib-ur==2025.1.1
+antlr4-python3-runtime==4.9.3
+firebase-admin==6.8.0
+jeepney==0.9.0
+tensorflow==2.18.0
+multipledispatch==1.0.0
+colorlover==0.3.0
+pynvml==12.0.0
+tsfresh==0.21.0
+polars==1.21.0
+imageio-ffmpeg==0.6.0
+cachetools==5.5.2
+bigframes==2.5.0
+rmm-cu12==25.2.0
+grpcio-status==1.71.0
+srsly==2.5.1
+sqlparse==0.5.3
+prompt_toolkit==3.0.51
+SecretStorage==3.3.3
+ucxx-cu12==0.42.0
+pathlib==1.0.1
+atpublic==5.1
+ipytree==0.2.2
+cudf-cu12==25.2.1
+pickleshare==0.7.5
+html5lib==1.1
+cvxopt==1.3.2
+opencv-python-headless==4.11.0.86
+idna==3.10
+docstring_parser==0.16
+google==2.0.3
+debugpy==1.8.0
+shapely==2.1.1
+google-genai==1.18.0
+spanner-graph-notebook==1.1.6
+smart-open==7.1.0
+weasel==0.4.1
+build==1.2.2.post1
+pydub==0.25.1
+pexpect==4.9.0
+colour==0.1.5
+soxr==0.5.0.post1
+PyGObject==3.42.0
+keras==3.8.0
+anyio==4.9.0
+jsonpatch==1.33
+jupyterlab_widgets==3.0.15
+slicer==0.0.8
+wurlitzer==3.1.1
+prometheus_client==0.22.1
+natsort==8.4.0
+google-cloud-firestore==2.21.0
+matplotlib-venn==1.1.2
+google-cloud-spanner==3.55.0
+lazy_loader==0.4
+google-generativeai==0.8.5
+nvidia-nvcomp-cu12==4.2.0.11
+folium==0.19.7
+greenlet==3.2.2
+dask==2024.12.1
+plotnine==0.14.5
+snowballstemmer==3.0.1
+gspread==6.2.1
+py-cpuinfo==9.0.0
+dask-cuda==25.2.0
+jieba==0.42.1
+importlib_metadata==8.7.0
+decorator==4.4.2
+markdown-it-py==3.0.0
+portpicker==1.5.2
+sympy==1.13.1
+libcuml-cu12==25.2.1
+tweepy==4.15.0
+easydict==1.13
+termcolor==3.1.0
+dlib==19.24.6
+py4j==0.10.9.7
+Send2Trash==1.8.3
+python-box==7.3.2
+huggingface-hub==0.32.4
+grpc-interceptor==0.15.4
+PySocks==1.7.1
+sphinxcontrib-htmlhelp==2.1.0
+bokeh==3.7.3
+beautifulsoup4==4.13.4
+SQLAlchemy==2.0.41
+nbclient==0.10.2
+glob2==0.7
+tensorflow-io-gcs-filesystem==0.37.1
+wrapt==1.17.2
+zict==3.0.0
+torchdata==0.11.0
+dopamine_rl==4.1.2
+threadpoolctl==3.6.0
+dask-cudf-cu12==25.2.2
+msgpack==1.1.0
+preshed==3.0.10
+ipython==7.34.0
+pydantic_core==2.33.2
+gdown==5.2.0
+google-cloud-language==2.17.1
+Bottleneck==1.4.2
+hf-xet==1.1.2
+spacy==3.8.7
+opt_einsum==3.4.0
+pyperclip==1.9.0
+propcache==0.3.1
+python-utils==3.9.1
+bigquery-magics==0.9.0
+defusedxml==0.7.1
+google-cloud-resource-manager==1.14.2
+pooch==1.8.2
+jax==0.5.2
+numba==0.60.0
+google-cloud-bigquery-connection==1.18.2
+narwhals==1.41.0
+blosc2==3.3.4
+pyshp==2.3.1
+sortedcontainers==2.4.0
+tensorflow-metadata==1.17.1
+jupyter_core==5.8.1
+opencv-contrib-python==4.11.0.86
+jsonschema-specifications==2025.4.1
+httpimport==1.4.1
+jupyter-server==1.16.0
+cramjam==2.10.0
+cuvs-cu12==25.2.1
+nbconvert==7.16.6
+scipy==1.15.3
+fastcore==1.7.29
+toolz==0.12.1
+cupy-cuda12x==13.3.0
+wcwidth==0.2.13
+umf==0.10.0
+xarray-einstats==0.9.0
+ndindex==1.10.0
+cvxpy==1.6.5
+contourpy==1.3.2
+ibis-framework==9.5.0
+ffmpy==0.6.0
+et_xmlfile==2.0.0
+cmake==3.31.6
+torchsummary==1.5.1
+distributed-ucxx-cu12==0.42.0
+mkl==2025.0.1
+webencodings==0.5.1
+Flask==3.1.1
+mistune==3.1.3
+ipyevents==2.0.2
+gym-notices==0.0.8
+cycler==0.12.1
+backports.tarfile==1.2.0
+langchain-text-splitters==0.3.8
+grpcio==1.72.1
+community==1.0.0b1
+humanize==4.12.3
+mpmath==1.3.0
+stumpy==1.13.0
+aiosignal==1.3.2
+grpclib==0.4.8
+iniconfig==2.1.0
+pyproject_hooks==1.2.0
+PyOpenGL==3.1.9
+vega-datasets==0.9.0
+referencing==0.36.2
+jupyter-console==6.1.0
+treescope==0.1.9
+sniffio==1.3.1
+attrs==25.3.0
+kaggle==1.7.4.5
+multidict==6.4.4
+bleach==6.2.0
+types-pytz==2025.2.0.20250516
+jaraco.classes==3.4.0
+langchain==0.3.25
+branca==0.8.1
+pyspark==3.5.1
+torchtune==0.6.1
+fastjsonschema==2.21.1
+pylibcudf-cu12==25.2.1
+traitlets==5.7.1
+docutils==0.21.2
+nibabel==5.3.2
+frozenlist==1.6.0
+imutils==0.5.4
+pycocotools==2.0.9
+kagglehub==0.3.12
+pyomo==6.9.2
+python-apt==0.0.0
+requirements-parser==0.9.0
+types-setuptools==80.9.0.20250529
+setuptools==75.2.0
+pip==24.1.2
+cryptography==3.4.8
+oauthlib==3.2.0
+pyparsing==2.4.7
+launchpadlib==1.10.16
+six==1.16.0
+distro==1.7.0
+PyGObject==3.42.1
+lazr.uri==1.0.6
+Mako==1.1.3
+blinker==1.4
+httplib2==0.20.2
+jeepney==0.7.1
+wadllib==1.3.6
+importlib-metadata==4.6.4
+Markdown==3.3.6
+SecretStorage==3.3.1
+MarkupSafe==2.0.1
+lazr.restfulclient==0.14.4
+python-apt==2.4.0+ubuntu4
+dbus-python==1.2.18
+PyJWT==2.3.0
+keyring==23.5.0
+zipp==1.0.0
+more-itertools==8.10.0
+jaraco.functools==4.0.1
+autocommand==2.2.2
+more-itertools==10.3.0
+tomli==2.0.1
+jaraco.collections==5.1.0
+wheel==0.43.0
+zipp==3.19.2
+importlib_metadata==8.0.0
+platformdirs==4.2.2
+packaging==24.1
+typing_extensions==4.12.2
+jaraco.context==5.3.0
+inflect==7.3.1
+importlib_resources==6.4.0
+typeguard==4.3.0
+jaraco.text==3.12.1
+backports.tarfile==1.2.0

wandb/run-20250612_232715-sam_single_cube/files/wandb-metadata.json

@@ -0,0 +1,46 @@
+{
+  "os": "Linux-6.1.123+-x86_64-with-glibc2.35",
+  "python": "CPython 3.11.13",
+  "startedAt": "2025-06-12T23:27:15.535692Z",
+  "args": [
+    "--dataset.repo_id=girardijp/sam_pick_cube_single",
+    "--policy.type=act",
+    "--output_dir=outputs/train/sam_single_cube",
+    "--job_name=sam_single_cube",
+    "--wandb.enable=true",
+    "--save_freq=600",
+    "--resume=false",
+    "--wandb.run_id=sam_single_cube",
+    "--steps=600"
+  ],
+  "program": "/root/lerobot/lerobot/scripts/train.py",
+  "root": "outputs/train/sam_single_cube",
+  "host": "a169ec209e92",
+  "executable": "/usr/bin/python3",
+  "cpu_count": 6,
+  "cpu_count_logical": 12,
+  "gpu": "NVIDIA A100-SXM4-40GB",
+  "gpu_count": 1,
+  "disk": {
+    "/": {
+      "total": "253055008768",
+      "used": "45656965120"
+    }
+  },
+  "memory": {
+    "total": "89631739904"
+  },
+  "cpu": {
+    "count": 6,
+    "countLogical": 12
+  },
+  "gpu_nvidia": [
+    {
+      "name": "NVIDIA A100-SXM4-40GB",
+      "memoryTotal": "42949672960",
+      "cudaCores": 6912,
+      "architecture": "Ampere"
+    }
+  ],
+  "cudaVersion": "12.4"
+}

wandb/run-20250612_232715-sam_single_cube/files/wandb-summary.json

@@ -0,0 +1 @@
+{"train/epochs":0.1756633119853614,"_runtime":95.128547223,"train/update_s":0.11615417072000354,"_step":600,"train/dataloading_s":0.0003153184600057557,"train/episodes":9.485818847209515,"_timestamp":1.749770928298207e+09,"train/grad_norm":75.18651704788208,"train/l1_loss":0.4601286053657532,"train/samples":4800,"train/kld_loss":0.20311912894248962,"train/lr":1.0000000000000021e-05,"train/loss":2.4739637327194215,"train/steps":600,"_wandb":{"runtime":95}}

wandb/run-20250612_232715-sam_single_cube/logs/debug-core.log

@@ -0,0 +1,14 @@
+{"time":"2025-06-12T23:27:15.479453991Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpq3cs_f8o/port-2971.txt","pid":2971,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
+{"time":"2025-06-12T23:27:15.491929187Z","level":"INFO","msg":"Will exit if parent process dies.","ppid":2971}
+{"time":"2025-06-12T23:27:15.496161577Z","level":"INFO","msg":"server is running","addr":{"IP":"127.0.0.1","Port":41515,"Zone":""}}
+{"time":"2025-06-12T23:27:15.52367636Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:27:15.538237413Z","level":"INFO","msg":"handleInformInit: received","streamId":"sam_single_cube","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:27:15.785991011Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"sam_single_cube","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:50.664113945Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:50.664196274Z","level":"INFO","msg":"connection: closing","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:50.664226877Z","level":"INFO","msg":"server is shutting down"}
+{"time":"2025-06-12T23:28:50.664339453Z","level":"INFO","msg":"connection: closed successfully","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:58.657919838Z","level":"ERROR","msg":"processOutgoingData: flush error","error":"write tcp 127.0.0.1:41515->127.0.0.1:59500: use of closed network connection","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:58.880202372Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:58.880219802Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"127.0.0.1:59500"}
+{"time":"2025-06-12T23:28:58.880236305Z","level":"INFO","msg":"server is closed"}

wandb/run-20250612_232715-sam_single_cube/logs/debug-internal.log

@@ -0,0 +1,15 @@
+{"time":"2025-06-12T23:27:15.538445428Z","level":"INFO","msg":"stream: starting","core version":"0.19.11","symlink path":"outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug-core.log"}
+{"time":"2025-06-12T23:27:15.785939368Z","level":"INFO","msg":"created new stream","id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.785984919Z","level":"INFO","msg":"stream: started","id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786041262Z","level":"INFO","msg":"sender: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786041501Z","level":"INFO","msg":"writer: Do: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:15.786093976Z","level":"INFO","msg":"handler: started","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:27:16.010724573Z","level":"INFO","msg":"Starting system monitor"}
+{"time":"2025-06-12T23:28:50.664224333Z","level":"INFO","msg":"stream: closing","id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:50.664258051Z","level":"INFO","msg":"Stopping system monitor"}
+{"time":"2025-06-12T23:28:50.664326487Z","level":"INFO","msg":"Stopped system monitor"}
+{"time":"2025-06-12T23:28:58.708086923Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-06-12T23:28:58.880071508Z","level":"INFO","msg":"handler: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.880113145Z","level":"INFO","msg":"writer: Close: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.880151088Z","level":"INFO","msg":"sender: closed","stream_id":"sam_single_cube"}
+{"time":"2025-06-12T23:28:58.8801821Z","level":"INFO","msg":"stream: closed","id":"sam_single_cube"}

wandb/run-20250612_232715-sam_single_cube/logs/debug.log

@@ -0,0 +1,23 @@
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Current SDK version is 0.19.11
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Configure stats pid to 2971
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from /root/.config/wandb/settings
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from /content/wandb/settings
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_setup.py:_flush():70] Loading settings from environment variables
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:setup_run_log_directory():724] Logging user logs to outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug.log
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:setup_run_log_directory():725] Logging internal logs to outputs/train/sam_single_cube/wandb/run-20250612_232715-sam_single_cube/logs/debug-internal.log
+2025-06-12 23:27:15,526 INFO    MainThread:2971 [wandb_init.py:init():852] calling init triggers
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():857] wandb.init called with sweep_config: {}
+config: {'dataset': {'repo_id': 'girardijp/sam_pick_cube_single', 'root': None, 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'tfs': {'brightness': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.8, 1.2]}}, 'contrast': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'contrast': [0.8, 1.2]}}, 'saturation': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'saturation': [0.5, 1.5]}}, 'hue': {'weight': 1.0, 'type': 'ColorJitter', 'kwargs': {'hue': [-0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'torchcodec'}, 'env': None, 'policy': {'type': 'act', 'n_obs_steps': 1, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'ACTION': <NormalizationMode.MEAN_STD: 'MEAN_STD'>}, 'input_features': {}, 'output_features': {}, 'device': 'cuda', 'use_amp': False, 'chunk_size': 100, 'n_action_steps': 100, 'vision_backbone': 'resnet18', 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'replace_final_stride_with_dilation': False, 'pre_norm': False, 'dim_model': 512, 'n_heads': 8, 'dim_feedforward': 3200, 'feedforward_activation': 'relu', 'n_encoder_layers': 4, 'n_decoder_layers': 1, 'use_vae': True, 'latent_dim': 32, 'n_vae_encoder_layers': 4, 'temporal_ensemble_coeff': None, 'dropout': 0.1, 'kl_weight': 10.0, 'optimizer_lr': 1e-05, 'optimizer_weight_decay': 0.0001, 'optimizer_lr_backbone': 1e-05}, 'output_dir': 'outputs/train/sam_single_cube', 'job_name': 'sam_single_cube', 'resume': False, 'seed': 1000, 'num_workers': 4, 'batch_size': 8, 'steps': 600, 'eval_freq': 20000, 'log_freq': 200, 'save_checkpoint': True, 'save_freq': 600, 'use_policy_training_preset': True, 'optimizer': {'type': 'adamw', 'lr': 1e-05, 'weight_decay': 0.0001, 'grad_clip_norm': 10.0, 'betas': [0.9, 0.999], 'eps': 1e-08}, 'scheduler': None, 'eval': {'n_episodes': 50, 'batch_size': 50, 'use_async_envs': False}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': 'sam_single_cube', 'mode': None}, '_wandb': {}}
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():893] starting backend
+2025-06-12 23:27:15,527 INFO    MainThread:2971 [wandb_init.py:init():897] sending inform_init request
+2025-06-12 23:27:15,535 INFO    MainThread:2971 [backend.py:_multiprocessing_setup():101] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2025-06-12 23:27:15,535 INFO    MainThread:2971 [wandb_init.py:init():907] backend started and connected
+2025-06-12 23:27:15,538 INFO    MainThread:2971 [wandb_init.py:init():1005] updated telemetry
+2025-06-12 23:27:15,539 INFO    MainThread:2971 [wandb_init.py:init():1029] communicating run to backend with 90.0 second timeout
+2025-06-12 23:27:16,005 INFO    MainThread:2971 [wandb_init.py:init():1104] starting run threads in backend
+2025-06-12 23:27:16,433 INFO    MainThread:2971 [wandb_run.py:_console_start():2573] atexit reg
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2421] redirect: wrap_raw
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2490] Wrapping output streams.
+2025-06-12 23:27:16,434 INFO    MainThread:2971 [wandb_run.py:_redirect():2513] Redirects installed.
+2025-06-12 23:27:16,439 INFO    MainThread:2971 [wandb_init.py:init():1150] run started, returning control to user process
+2025-06-12 23:28:50,663 INFO    MsgRouterThr:2971 [mailbox.py:close():129] [no run ID] Closing mailbox, abandoning 2 handles.

wandb/run-20250612_232715-sam_single_cube/run-sam_single_cube.wandb

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