exported/dreamer_manifest.json

{
  "tensors": {
    "deter_init_w": {
      "offset": 0,
      "length": 512,
      "shape": [
        1,
        512
      ]
    },
    "gru_norm_b": {
      "offset": 6293504,
      "length": 1536,
      "shape": [
        1536
      ]
    },
    "gru_norm_w": {
      "offset": 6299648,
      "length": 1536,
      "shape": [
        1536
      ]
    },
    "img_in_w": {
      "offset": 6305792,
      "length": 524800,
      "shape": [
        512,
        1025
      ]
    },
    "img_in_norm_b": {
      "offset": 8404992,
      "length": 512,
      "shape": [
        512
      ]
    },
    "img_in_norm_w": {
      "offset": 8407040,
      "length": 512,
      "shape": [
        512
      ]
    },
    "img_out_w": {
      "offset": 8409088,
      "length": 262144,
      "shape": [
        512,
        512
      ]
    },
    "img_out_norm_b": {
      "offset": 9457664,
      "length": 512,
      "shape": [
        512
      ]
    },
    "img_out_norm_w": {
      "offset": 9459712,
      "length": 512,
      "shape": [
        512
      ]
    },
    "imgs_stat_b": {
      "offset": 9461760,
      "length": 1024,
      "shape": [
        1024
      ]
    },
    "imgs_stat_w": {
      "offset": 9465856,
      "length": 524288,
      "shape": [
        1024,
        512
      ]
    },
    "obs_out_w": {
      "offset": 11563008,
      "length": 524288,
      "shape": [
        512,
        1024
      ]
    },
    "obs_out_norm_b": {
      "offset": 13660160,
      "length": 512,
      "shape": [
        512
      ]
    },
    "obs_out_norm_w": {
      "offset": 13662208,
      "length": 512,
      "shape": [
        512
      ]
    },
    "obs_stat_b": {
      "offset": 13664256,
      "length": 1024,
      "shape": [
        1024
      ]
    },
    "obs_stat_w": {
      "offset": 13668352,
      "length": 524288,
      "shape": [
        1024,
        512
      ]
    },
    "enc_linear0_w": {
      "offset": 15765504,
      "length": 36864,
      "shape": [
        512,
        72
      ]
    },
    "enc_linear1_w": {
      "offset": 15912960,
      "length": 262144,
      "shape": [
        512,
        512
      ]
    },
    "enc_linear2_w": {
      "offset": 16961536,
      "length": 262144,
      "shape": [
        512,
        512
      ]
    },
    "enc_norm0_b": {
      "offset": 18010112,
      "length": 512,
      "shape": [
        512
      ]
    },
    "enc_norm0_w": {
      "offset": 18012160,
      "length": 512,
      "shape": [
        512
      ]
    },
    "enc_norm1_b": {
      "offset": 18014208,
      "length": 512,
      "shape": [
        512
      ]
    },
    "enc_norm1_w": {
      "offset": 18016256,
      "length": 512,
      "shape": [
        512
      ]
    },
    "enc_norm2_b": {
      "offset": 18018304,
      "length": 512,
      "shape": [
        512
      ]
    },
    "enc_norm2_w": {
      "offset": 18020352,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_linear0_w": {
      "offset": 18022400,
      "length": 786432,
      "shape": [
        512,
        1536
      ]
    },
    "dec_linear1_w": {
      "offset": 21168128,
      "length": 262144,
      "shape": [
        512,
        512
      ]
    },
    "dec_linear2_w": {
      "offset": 22216704,
      "length": 262144,
      "shape": [
        512,
        512
      ]
    },
    "dec_norm0_b": {
      "offset": 23265280,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_norm0_w": {
      "offset": 23267328,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_norm1_b": {
      "offset": 23269376,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_norm1_w": {
      "offset": 23271424,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_norm2_b": {
      "offset": 23273472,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_norm2_w": {
      "offset": 23275520,
      "length": 512,
      "shape": [
        512
      ]
    },
    "dec_out_b": {
      "offset": 23277568,
      "length": 72,
      "shape": [
        72
      ]
    },
    "dec_out_w": {
      "offset": 23277856,
      "length": 36864,
      "shape": [
        72,
        512
      ]
    },
    "gru_w": {
      "offset": 2048,
      "length": 1572864,
      "shape": [
        1536,
        1024
      ]
    }
  }
}