{
  "Diffusion": true,
  "S": 1024,
  "T": 12,
  "action_contrastive_loss": false,
  "action_domains": [
    "language_table",
    "ucsd_pick_and_place_dataset_converted_externally_to_rlds",
    "kaist_nonprehensile_converted_externally_to_rlds",
    "ucsd_kitchen_dataset_converted_externally_to_rlds",
    "utokyo_xarm_bimanual_converted_externally_to_rlds",
    "stanford_hydra_dataset_converted_externally_to_rlds",
    "austin_sirius_dataset_converted_externally_to_rlds",
    "berkeley_fanuc_manipulation",
    "berkeley_mvp_converted_externally_to_rlds",
    "berkeley_rpt_converted_externally_to_rlds",
    "cmu_play_fusion",
    "iamlab_cmu_pickup_insert_converted_externally_to_rlds",
    "qut_dexterous_manpulation",
    "robo_net",
    "dlr_sara_grid_clamp_converted_externally_to_rlds",
    "cmu_stretch",
    "columbia_cairlab_pusht_real",
    "droid",
    "toto",
    "io_ai_tech",
    "dobbe",
    "berkeley_gnm_cory_hall",
    "plex_robosuite",
    "berkeley_cable_routing",
    "imperial_wrist_dataset",
    "bc_z",
    "kuka",
    "roboturk",
    "robomimic",
    "nyu_door_opening_surprising_effectiveness"
  ],
  "action_loss_weight": 1.0,
  "action_network": "modulate",
  "action_stats": [
    [
      [
        0.00014842326345387846,
        -0.0005635050474666059
      ],
      [
        0.030163198709487915,
        0.042305462062358856
      ]
    ],
    [
      [
        0.14697618782520294,
        -0.12370482087135315,
        0.051797714084386826,
        -0.07087255269289017
      ],
      [
        0.48446422815322876,
        0.4629215896129608,
        0.5406527519226074,
        0.8932808637619019
      ]
    ],
    [
      [
        0.0019365031039342284,
        0.00024300716177094728,
        0.0008035349892452359,
        -0.0021898974664509296,
        4.0033126424532384e-05,
        -0.0037320367991924286,
        181.3382110595703,
        170.906005859375,
        186.00167846679688,
        153.71453857421875,
        174.13552856445312,
        83.48700714111328,
        32.40367889404297,
        1.0105239152908325,
        0.70769864320755,
        0.33714282512664795,
        0.4880707859992981,
        0.49914246797561646,
        0.7389975786209106,
        1.229773998260498
      ],
      [
        0.014657468535006046,
        0.016509365290403366,
        0.012914449907839298,
        0.023157890886068344,
        0.020476851612329483,
        0.019903959706425667,
        27.33956527709961,
        34.60658645629883,
        25.291311264038086,
        53.12702941894531,
        34.06013870239258,
        50.88619613647461,
        33.19802474975586,
        0.5348116159439087,
        0.4462398290634155,
        0.13059484958648682,
        0.37823718786239624,
        0.3443543016910553,
        0.4008789658546448,
        0.5644641518592834
      ]
    ],
    [
      [
        408.48406982421875,
        118.63397979736328,
        198.41452026367188,
        -121.96654510498047,
        -34.39997863769531,
        52.22698974609375,
        0.7438188791275024,
        0.038725052028894424
      ],
      [
        122.93132019042969,
        107.72244262695312,
        128.7881317138672,
        115.888916015625,
        27.235536575317383,
        40.505306243896484,
        0.43652451038360596,
        0.1929423063993454
      ]
    ],
    [
      [
        0.46141278743743896,
        0.10542168468236923,
        0.25353577733039856,
        -1.685599684715271,
        -0.05627294257283211,
        -0.4933978319168091,
        0.34305593371391296,
        0.44558027386665344,
        0.5123444199562073,
        0.2677648663520813,
        1.277772307395935,
        0.12375026196241379,
        0.11488401144742966,
        0.33085882663726807
      ],
      [
        0.053879205137491226,
        0.05707748979330063,
        0.04467933997511864,
        2.469133138656616,
        0.18709321320056915,
        0.22889232635498047,
        0.47383636236190796,
        0.06301931291818619,
        0.041924599558115005,
        0.047896090894937515,
        2.701061248779297,
        0.2972114384174347,
        0.7424992918968201,
        0.4667799174785614
      ]
    ],
    [
      [
        0.0007764044567011297,
        0.0001343307230854407,
        -0.00026648343191482127,
        0.0013218839885666966,
        -0.004740390460938215,
        0.002773461164906621,
        0.5106820464134216
      ],
      [
        0.008042743429541588,
        0.00913731288164854,
        0.009599598124623299,
        0.04121660068631172,
        0.038332853466272354,
        0.04602774232625961,
        0.4999658763408661
      ]
    ],
    [
      [
        0.07727599143981934,
        0.03225162252783775,
        0.04257211461663246,
        0.0,
        0.0,
        -0.01612210087478161,
        0.13071605563163757
      ],
      [
        0.3917539417743683,
        0.30044373869895935,
        0.27837157249450684,
        0.0,
        0.0,
        0.081514872610569,
        0.9911611676216125
      ]
    ],
    [
      [
        0.0007766556227579713,
        -0.000321519240969792,
        -0.0014813995221629739,
        -0.0007485907408408821,
        -0.00015667964180465788,
        0.0001845337392296642
      ],
      [
        0.003409236203879118,
        0.004994169808924198,
        0.005332312546670437,
        0.007559089455753565,
        0.004051606170833111,
        0.008588160388171673
      ]
    ],
    [
      [
        -6.743222911609337e-05,
        0.0031809681095182896,
        -0.00013550207950174809,
        -0.0009742751135490835,
        -8.3738968896796e-06,
        -0.002912015886977315,
        -0.0006995691219344735,
        0.48066604137420654
      ],
      [
        0.002549938391894102,
        0.012658610939979553,
        0.005411175545305014,
        0.018054410815238953,
        0.0016273874789476395,
        0.021100502461194992,
        0.005715933162719011,
        0.4996056854724884
      ]
    ],
    [
      [
        0.00014941584959160537,
        -0.00028024936909787357,
        -8.037472071009688e-06,
        -0.00032872759038582444,
        1.9844068447127938e-05,
        3.272057801950723e-05,
        8.096991950878873e-05,
        0.4784493148326874
      ],
      [
        0.0015258367639034986,
        0.004546448588371277,
        0.0007782428874634206,
        0.003019175725057721,
        0.0010663573630154133,
        0.005132743623107672,
        0.004171756561845541,
        0.4998187720775604
      ]
    ],
    [
      [
        0.000523168477229774,
        3.85410530725494e-05,
        -0.00017000196385197341,
        -0.00029378157341852784,
        -0.00036922883009538054,
        -0.0001573827030370012,
        5.717058229492977e-06,
        0.5699702501296997,
        0.002427969593554735
      ],
      [
        0.0018014844972640276,
        0.002389610279351473,
        0.0018651892896741629,
        0.039326585829257965,
        0.03775598481297493,
        0.005358702037483454,
        0.007674811407923698,
        0.49466049671173096,
        0.04917134344577789
      ]
    ],
    [
      [
        0.5280895829200745,
        0.02888699807226658,
        0.18680934607982635,
        -0.01308287400752306,
        0.9998903870582581,
        0.003612307133153081,
        0.016001908108592033,
        0.5531076192855835
      ],
      [
        0.08082365244626999,
        0.11135152727365494,
        0.07754139602184296,
        0.01604105904698372,
        0.0006265711272135377,
        0.007812995463609695,
        0.013805469498038292,
        0.497190922498703
      ]
    ],
    [
      [
        3.650687176559586e-06,
        -0.000507326505612582,
        -0.00031988348928280175,
        0.000982428900897503,
        -3.8030557334423065e-05,
        -0.002412878442555666,
        0.004487304482609034,
        -0.0035084427800029516
      ],
      [
        0.012881123460829258,
        0.015523286536335945,
        0.012135118246078491,
        0.0009839057456701994,
        0.0022208373993635178,
        0.024184072390198708,
        0.9997038841247559,
        0.027473121881484985
      ]
    ],
    [
      [
        -1.2117172445869073e-05,
        1.6497699107276276e-05,
        -0.008070996962487698,
        -3.756756632355973e-05,
        -0.02855828031897545
      ],
      [
        0.03243051841855049,
        0.03239370137453079,
        0.09075836837291718,
        0.17676453292369843,
        0.9996473789215088
      ]
    ],
    [
      [
        -1.662798604229465e-05,
        -4.2423445847816765e-05,
        -0.00039035530062392354,
        2.9382475986494683e-05,
        9.349627362098545e-05,
        7.735285180388018e-05,
        1.0
      ],
      [
        0.0004397016600705683,
        0.0005161615554243326,
        0.0012931948294863105,
        0.0005689726676791906,
        0.0007594820926897228,
        0.0007169033051468432,
        0.0
      ]
    ],
    [
      [
        0.0003590668202377856,
        0.0,
        0.001637771725654602,
        0.0,
        0.0,
        0.0,
        0.39824214577674866,
        0.0054051512852311134
      ],
      [
        0.004081381484866142,
        0.0,
        0.003803750965744257,
        0.0,
        0.0,
        0.0,
        0.48956871032714844,
        0.07332666963338852
      ]
    ],
    [
      [
        0.0,
        0.0,
        0.0,
        0.0,
        0.009795918129384518,
        -0.0013581214006990194,
        0.0016117944614961743,
        0.0
      ],
      [
        0.0,
        0.0,
        0.0,
        0.0,
        0.09848489612340927,
        0.013396660797297955,
        0.01613754965364933,
        0.0
      ]
    ],
    [
      [
        0.5393196940422058,
        0.0013354304246604443,
        0.3156941831111908,
        0.3164699971675873,
        -0.09018929302692413,
        -0.049339085817337036,
        0.40983372926712036
      ],
      [
        0.11741136759519577,
        0.17491821944713593,
        0.16181626915931702,
        2.7440731525421143,
        0.3496827483177185,
        0.7598394155502319,
        0.43021807074546814
      ]
    ],
    [
      [
        0.0,
        -0.6654278039932251,
        0.1888580173254013,
        0.03210142254829407,
        0.00612324383109808,
        0.38309070467948914,
        0.009382354095578194,
        0.3636060655117035
      ],
      [
        0.0,
        0.5778681635856628,
        0.296415776014328,
        0.3222154378890991,
        0.07795067131519318,
        0.12293250113725662,
        0.19515110552310944,
        0.10152395814657211
      ]
    ],
    [
      [
        2.957821561722085e-05,
        0.00012851174687966704,
        -0.00010689908231142908,
        -5.97012804064434e-05,
        0.00022397778229787946,
        6.999688048381358e-05,
        0.09176551550626755
      ],
      [
        0.0028307351749390364,
        0.002651946386322379,
        0.0025581379886716604,
        0.018357520923018456,
        0.025164088234305382,
        0.024045433849096298,
        0.5909407734870911
      ]
    ],
    [
      [
        -0.00011003677354892716,
        0.001112840254791081,
        -0.00011267208174103871,
        -7.512857700930908e-05,
        -0.0006745870341546834,
        -5.703312126570381e-05,
        0.6326711177825928
      ],
      [
        0.043496448546648026,
        0.04464876651763916,
        0.12467490881681442,
        0.005452098790556192,
        0.011218013241887093,
        0.00624604569748044,
        0.39724212884902954
      ]
    ],
    [
      [
        0.06122741475701332,
        0.0038701200392097235
      ],
      [
        0.025848353281617165,
        0.0030985879711806774
      ]
    ],
    [
      [
        0.060740936547517776,
        0.053044628351926804,
        -0.04193497821688652,
        -0.000676018709782511,
        -0.0015231040306389332,
        0.004273010417819023,
        -0.05146767199039459
      ],
      [
        0.34811627864837646,
        0.46360549330711365,
        0.4386604428291321,
        0.023519689217209816,
        0.019431674852967262,
        0.1616460531949997,
        0.9985936880111694
      ]
    ],
    [
      [
        0.0,
        0.0,
        0.049097269773483276,
        0.07730317115783691,
        -0.07240438461303711,
        0.02373087964951992,
        0.10240031778812408
      ],
      [
        0.0,
        0.0,
        0.3465680181980133,
        0.2670310139656067,
        0.1823672205209732,
        0.1818883866071701,
        0.21153412759304047
      ]
    ],
    [
      [
        0.0003212452866137028,
        -0.0010083492379635572,
        0.00092211680021137,
        0.001238797907717526,
        -4.7416866436833516e-05,
        2.5170325898216106e-05,
        0.5775114297866821,
        0.023655574768781662
      ],
      [
        0.003074005013331771,
        0.0067475223913788795,
        0.010976199060678482,
        0.024110153317451477,
        0.003232581540942192,
        0.0039499602280557156,
        0.49393802881240845,
        0.15197230875492096
      ]
    ],
    [
      [
        0.00028649900923483074,
        -0.008722408674657345,
        -0.03069918043911457,
        -0.0008381816442124546,
        -0.016971644014120102,
        -0.05745099112391472,
        -0.0026707653887569904,
        -0.024192843586206436,
        -0.07967454195022583,
        -0.004741811193525791,
        -0.030430495738983154,
        -0.09769809991121292,
        -0.006405732128769159,
        -0.03590046241879463,
        -0.11270859092473984,
        -0.007021840196102858,
        -0.04052259027957916,
        -0.12620966136455536,
        -0.006953817792236805,
        -0.04445187374949455,
        -0.13877084851264954,
        -0.006491991225630045,
        -0.04791347682476044,
        -0.15081816911697388,
        -0.0057747503742575645,
        -0.05111181363463402,
        -0.16245798766613007,
        -0.004867491777986288,
        -0.054257530719041824,
        -0.1738300770521164,
        0.1657339185476303,
        0.15363934636116028,
        0.14477591216564178,
        0.13839827477931976,
        0.14092908799648285,
        0.15468865633010864,
        0.16648422181606293,
        0.17608821392059326,
        0.1841760128736496,
        0.19062727689743042,
        -0.00996700394898653,
        0.0009040668956004083,
        0.004995268769562244,
        -0.018695320934057236,
        0.0023894852492958307,
        0.009505861438810825,
        -0.025692706927657127,
        0.0043935589492321014,
        0.013725746423006058,
        -0.031206561252474785,
        0.006276523228734732,
        0.017453059554100037,
        -0.03552345186471939,
        0.00730851711705327,
        0.0201703030616045,
        -0.03902909904718399,
        0.0068913171999156475,
        0.021356917917728424,
        -0.04216034710407257,
        0.005601761396974325,
        0.021485209465026855,
        -0.04517875239253044,
        0.0038381200283765793,
        0.020964166149497032,
        -0.04812570661306381,
        0.0018662408692762256,
        0.020047230646014214,
        -0.05107533559203148,
        -0.00014347363321576267,
        0.01891058310866356
      ],
      [
        0.04161107912659645,
        0.04643801972270012,
        0.07714500278234482,
        0.06882365792989731,
        0.07853177934885025,
        0.13730594515800476,
        0.08874603360891342,
        0.10269544273614883,
        0.18733122944831848,
        0.10394992679357529,
        0.12144353240728378,
        0.228963240981102,
        0.11596192419528961,
        0.1365172266960144,
        0.26162227988243103,
        0.1259639710187912,
        0.1488255262374878,
        0.2899230718612671,
        0.13448232412338257,
        0.1590842306613922,
        0.3123351037502289,
        0.141754150390625,
        0.16746656596660614,
        0.3306305706501007,
        0.1480536013841629,
        0.17442390322685242,
        0.34733083844184875,
        0.1536930501461029,
        0.18031899631023407,
        0.36258259415626526,
        0.36356091499328613,
        0.3559049963951111,
        0.34951725602149963,
        0.3456125855445862,
        0.3479859530925751,
        0.35668134689331055,
        0.36377906799316406,
        0.3706970512866974,
        0.3976169526576996,
        0.39772698283195496,
        0.03057853877544403,
        0.02315731719136238,
        0.020660309121012688,
        0.054019346833229065,
        0.039159927517175674,
        0.03595462813973427,
        0.07279643416404724,
        0.051145341247320175,
        0.04803183302283287,
        0.08770721405744553,
        0.060230545699596405,
        0.05767446011304855,
        0.0995459109544754,
        0.06725169718265533,
        0.06545353680849075,
        0.10892871767282486,
        0.07302306592464447,
        0.07213838398456573,
        0.116375632584095,
        0.0778125748038292,
        0.07771806418895721,
        0.12312348932027817,
        0.0816049873828888,
        0.08206423372030258,
        0.1288398653268814,
        0.08456701040267944,
        0.08530736714601517,
        0.13414840400218964,
        0.08684109151363373,
        0.0876062661409378
      ]
    ],
    [
      [
        0.0,
        0.0,
        0.0,
        0.06579020619392395,
        0.0,
        0.0,
        -0.0417020283639431,
        0.05600078031420708,
        0.8763857483863831,
        0.0,
        -0.0006691364105790854,
        0.0005162839079275727,
        -0.0025432889815419912
      ],
      [
        0.0,
        0.0,
        0.0,
        0.3762088716030121,
        0.0,
        0.0,
        0.15823891758918762,
        0.2252153754234314,
        0.31409522891044617,
        0.0,
        0.023507647216320038,
        0.036004047840833664,
        0.05768127366900444
      ]
    ],
    [
      [
        -0.15657226741313934,
        0.00228637782856822,
        -0.0009536752477288246,
        -0.00012742729450110346,
        0.0,
        0.0014414743054658175,
        -0.0015724773984402418,
        -0.0011747290845960379
      ],
      [
        0.9879051446914673,
        0.09651217609643936,
        0.08441831171512604,
        0.06647706776857376,
        0.0,
        0.0495850145816803,
        0.06368337571620941,
        0.06135875731706619
      ]
    ],
    [
      [
        0.17389154434204102,
        0.005625918973237276,
        -0.1695142686367035,
        0.0031083673238754272,
        0.005127986893057823,
        0.012693661265075207,
        -0.4065398871898651
      ],
      [
        0.2601781189441681,
        0.13021306693553925,
        0.4979441463947296,
        0.022246459499001503,
        0.06382154673337936,
        0.08343781530857086,
        0.913625180721283
      ]
    ],
    [
      [
        0.02290557324886322,
        -0.00010951685544569045,
        -0.011411379091441631,
        -0.0015635089948773384,
        0.04783362150192261,
        -0.0063293022103607655,
        0.0013472747523337603,
        0.001141763525083661
      ],
      [
        0.09001470357179642,
        0.00812098290771246,
        0.033615339547395706,
        0.013250669464468956,
        0.21339112520217896,
        0.01224832609295845,
        0.019686469808220863,
        0.00791964028030634
      ]
    ]
  ],
  "action_token_size": 64,
  "arch": "STTransformerDecoder",
  "attn_drop": 0.1,
  "attn_dropout": 0.1,
  "buffer_size": 64,
  "d_action": 28,
  "d_actions": [
    2,
    4,
    100,
    8,
    14,
    35,
    70,
    6,
    16,
    120,
    18,
    80,
    8,
    5,
    7,
    8,
    40,
    49,
    120,
    7,
    7,
    2,
    7,
    35,
    8,
    350,
    13,
    40,
    21,
    8
  ],
  "d_model": 256,
  "dataloader_apply_corruption": false,
  "dataloader_apply_mask": true,
  "dataloader_mask_ratio_min": 0.1,
  "diffloss_d": 4,
  "diffloss_w": 1024,
  "diffusion_batch_mul": 1,
  "dim": 512,
  "drop_action_ratio": 0.0,
  "factored_vocab_size": 512,
  "grad_checkpointing": false,
  "image_vocab_size": null,
  "init_actions": true,
  "jointly_predict_actions": false,
  "jointly_predict_states": true,
  "label_drop_prob": 0.5,
  "mask_ratio_min": 0.7,
  "maskgit_steps": 16,
  "max_corrupt_rate": 0.2,
  "mlp_bias": false,
  "mlp_drop": 0.05,
  "mlp_ratio": 4.0,
  "non_mlm_ratio": 0.2,
  "num_factored_vocabs": 2,
  "num_heads": 8,
  "num_layers": 32,
  "num_prompt_frames": 4,
  "num_sampling_steps": "100",
  "patch_size": 2,
  "predict_unmask": false,
  "proj_bias": true,
  "proj_dropout": 0.1,
  "qk_norm": false,
  "qkv_bias": true,
  "random_dummy_action": true,
  "shared_action_mlps": true,
  "use_actions": true,
  "use_mup": false,
  "vae_embed_dim": 4,
  "vae_stride": 1
}