| { |
| "dataset": { |
| "repo_id": "roboeval_ee_delta_20hz_v3_merged", |
| "root": "/weka/helenw/data/huggingface/lerobot/roboeval_ee_delta_20hz_v3_merged", |
| "episodes": [ |
| 1887, |
| 1888, |
| 1889, |
| 1890, |
| 1891, |
| 1892, |
| 1893, |
| 1894, |
| 1895, |
| 1896, |
| 1897, |
| 1898, |
| 1899, |
| 1900, |
| 1901, |
| 1902, |
| 1903, |
| 1904, |
| 1905, |
| 1906, |
| 1907, |
| 1908, |
| 1909, |
| 1910, |
| 1911, |
| 1912, |
| 1913, |
| 1914, |
| 1915, |
| 1916, |
| 1917, |
| 1918, |
| 1919, |
| 1920, |
| 1921, |
| 1922, |
| 1923, |
| 1924, |
| 1925, |
| 1926, |
| 1927, |
| 1928, |
| 1929, |
| 1930, |
| 1931, |
| 1932, |
| 1933, |
| 1934, |
| 1935, |
| 1936, |
| 1937, |
| 1938, |
| 1939, |
| 1940, |
| 1941, |
| 1942, |
| 1943, |
| 1944, |
| 1945, |
| 1946, |
| 1947, |
| 1948, |
| 1949, |
| 1950, |
| 1951, |
| 1952, |
| 1953, |
| 1954, |
| 1955, |
| 1956, |
| 1957, |
| 1958, |
| 1959, |
| 1960, |
| 1961, |
| 1962, |
| 1963, |
| 1964, |
| 1965, |
| 1966, |
| 1967, |
| 1968, |
| 1969, |
| 1970, |
| 1971, |
| 1972, |
| 1973, |
| 1974, |
| 1975, |
| 1976, |
| 1977, |
| 1978, |
| 1979, |
| 1980, |
| 1981, |
| 1982, |
| 1983, |
| 1984, |
| 1985, |
| 1986, |
| 1987, |
| 1988, |
| 1989, |
| 1990, |
| 1991, |
| 1992, |
| 1993, |
| 1994, |
| 1995, |
| 1996, |
| 1997, |
| 1998, |
| 1999, |
| 2000, |
| 2001, |
| 2002, |
| 2003, |
| 2004, |
| 2005, |
| 2006, |
| 2007, |
| 2008, |
| 2009, |
| 2010, |
| 2011, |
| 2012, |
| 2013, |
| 2014, |
| 2015, |
| 2016, |
| 2017, |
| 2018, |
| 2019, |
| 2020, |
| 2021, |
| 2022, |
| 2023, |
| 2024, |
| 2025, |
| 2026, |
| 2027, |
| 2028, |
| 2029, |
| 2030, |
| 2031, |
| 2032, |
| 2033, |
| 2034, |
| 2035, |
| 2036, |
| 2037, |
| 2038, |
| 2039, |
| 2040, |
| 2041, |
| 2042, |
| 2043, |
| 2044, |
| 2045, |
| 2046, |
| 2047, |
| 2048, |
| 2049, |
| 2050, |
| 2051, |
| 2052, |
| 2053, |
| 2054, |
| 2055, |
| 2056, |
| 2057, |
| 2058, |
| 2059, |
| 2060, |
| 2061, |
| 2062, |
| 2063, |
| 2064, |
| 2065, |
| 2066, |
| 2067, |
| 2068, |
| 2069, |
| 2070, |
| 2071, |
| 2072, |
| 2073, |
| 2074, |
| 2075, |
| 2076, |
| 2077, |
| 2078, |
| 2079, |
| 2080, |
| 2081, |
| 2082, |
| 2083, |
| 2084, |
| 2085, |
| 2086, |
| 2087, |
| 2088, |
| 2089, |
| 2090, |
| 2091, |
| 2092, |
| 2093, |
| 2094, |
| 2095, |
| 2096, |
| 2097, |
| 2098, |
| 2099, |
| 2100, |
| 2101, |
| 2102, |
| 2103, |
| 2104, |
| 2105, |
| 2106, |
| 2107, |
| 2108, |
| 2109, |
| 2110, |
| 2111, |
| 2112, |
| 2113, |
| 2114, |
| 2115, |
| 2116, |
| 2117, |
| 2118, |
| 2119, |
| 2120, |
| 2121, |
| 2122, |
| 2123, |
| 2124, |
| 2125, |
| 2126, |
| 2127, |
| 2128, |
| 2129, |
| 2130, |
| 2131, |
| 2132, |
| 2133, |
| 2134, |
| 2135, |
| 2136, |
| 2137, |
| 2138, |
| 2139, |
| 2140, |
| 2141, |
| 2142, |
| 2143, |
| 2144, |
| 2145, |
| 2146, |
| 2147, |
| 2148, |
| 2149, |
| 2150, |
| 2151, |
| 2152, |
| 2153, |
| 2154, |
| 2155, |
| 2156, |
| 2157, |
| 2158, |
| 2159, |
| 2160, |
| 2161, |
| 2162, |
| 2163, |
| 2164, |
| 2165, |
| 2166, |
| 2167, |
| 2168, |
| 2169, |
| 2170, |
| 2171, |
| 2172, |
| 2173, |
| 2174, |
| 2175, |
| 2176, |
| 2177, |
| 2178, |
| 2179, |
| 2180, |
| 2181, |
| 2182, |
| 2183, |
| 2184, |
| 2185, |
| 2186, |
| 2187, |
| 2188, |
| 2189, |
| 2190, |
| 2191, |
| 2192, |
| 2193, |
| 2194, |
| 2195, |
| 2196, |
| 2197, |
| 2198, |
| 2199, |
| 2200, |
| 2201, |
| 2202, |
| 2203, |
| 2204, |
| 2205, |
| 2206, |
| 2207, |
| 2208, |
| 2209, |
| 2210, |
| 2211, |
| 2212, |
| 2213, |
| 2214, |
| 2215, |
| 2216, |
| 2217, |
| 2218, |
| 2219, |
| 2220, |
| 2221, |
| 2222, |
| 2223, |
| 2224, |
| 2225, |
| 2226, |
| 2227, |
| 2228, |
| 2229, |
| 2230, |
| 2231, |
| 2232, |
| 2233, |
| 2234, |
| 2235 |
| ], |
| "image_transforms": { |
| "enable": true, |
| "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 |
| ] |
| } |
| }, |
| "affine": { |
| "weight": 1.0, |
| "type": "RandomAffine", |
| "kwargs": { |
| "degrees": [ |
| -5.0, |
| 5.0 |
| ], |
| "translate": [ |
| 0.05, |
| 0.05 |
| ] |
| } |
| } |
| } |
| }, |
| "revision": "local", |
| "use_imagenet_stats": true, |
| "video_backend": "torchcodec", |
| "streaming": false |
| }, |
| "env": null, |
| "policy": { |
| "type": "act", |
| "n_obs_steps": 1, |
| "input_features": { |
| "observation.images.cam_head": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.images.cam_left_wrist": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.images.cam_right_wrist": { |
| "type": "VISUAL", |
| "shape": [ |
| 3, |
| 256, |
| 256 |
| ] |
| }, |
| "observation.state": { |
| "type": "STATE", |
| "shape": [ |
| 16 |
| ] |
| } |
| }, |
| "output_features": { |
| "action": { |
| "type": "ACTION", |
| "shape": [ |
| 14 |
| ] |
| } |
| }, |
| "device": "cuda", |
| "use_amp": false, |
| "use_peft": false, |
| "push_to_hub": false, |
| "repo_id": null, |
| "private": null, |
| "tags": null, |
| "license": null, |
| "pretrained_path": null, |
| "chunk_size": 16, |
| "n_action_steps": 8, |
| "normalization_mapping": { |
| "VISUAL": "MEAN_STD", |
| "STATE": "QUANTILES", |
| "ACTION": "QUANTILES" |
| }, |
| "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": 0.01, |
| "optimizer_lr": 0.0001, |
| "optimizer_weight_decay": 0.0001, |
| "optimizer_lr_backbone": 1e-05 |
| }, |
| "output_dir": "/weka/helenw/outputs/act_RotateValve", |
| "job_name": "act_chunk16_roboeval_v3_merged", |
| "resume": false, |
| "seed": 1000, |
| "num_workers": 2, |
| "batch_size": 512, |
| "steps": 5000, |
| "eval_freq": 20000, |
| "log_freq": 10, |
| "tolerance_s": 0.0001, |
| "save_checkpoint": true, |
| "save_freq": 1000, |
| "use_policy_training_preset": true, |
| "optimizer": { |
| "type": "adamw", |
| "lr": 0.0001, |
| "weight_decay": 0.0001, |
| "grad_clip_norm": 10.0, |
| "betas": [ |
| 0.9, |
| 0.999 |
| ], |
| "eps": 1e-08 |
| }, |
| "scheduler": { |
| "type": "diffuser", |
| "num_warmup_steps": 500, |
| "name": "cosine" |
| }, |
| "eval": { |
| "n_episodes": 50, |
| "batch_size": 50, |
| "use_async_envs": false |
| }, |
| "wandb": { |
| "enable": true, |
| "disable_artifact": true, |
| "project": "lerobot", |
| "entity": null, |
| "notes": null, |
| "run_id": "zoaxuabk", |
| "mode": null |
| }, |
| "peft": null, |
| "use_rabc": false, |
| "rabc_progress_path": null, |
| "rabc_kappa": 0.01, |
| "rabc_epsilon": 1e-06, |
| "rabc_head_mode": "sparse", |
| "rename_map": {}, |
| "checkpoint_path": null |
| } |
