{
  "status": "pass",
  "checkpoint": "state/checkpoints/astronaut-w4both-w4a16-default50-early/step-001-of-049.pt",
  "tensor_count": 108,
  "tensors": [
    {
      "path": "root.state.timesteps",
      "shape": [
        49
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0,
      "max": 0.800000011920929,
      "mean": 0.14573542773723602
    },
    {
      "path": "root.state.latents",
      "shape": [
        16049,
        96
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": -5.126751899719238,
      "max": 4.734387397766113,
      "mean": -0.014345129020512104
    },
    {
      "path": "root.state.audio_latents",
      "shape": [
        414,
        32
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": -3.742875337600708,
      "max": 4.066157817840576,
      "mean": -0.015999415889382362
    },
    {
      "path": "root.state.prompt_embeds",
      "shape": [
        1,
        5585,
        5120
      ],
      "dtype": "torch.bfloat16",
      "finite": true,
      "min": -160.0,
      "max": 14336.0,
      "mean": 0.006581898313015699
    },
    {
      "path": "root.state.row_timestep_plan[0][0]",
      "shape": [
        2
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0,
      "max": 0.9990000128746033,
      "mean": 0.49950000643730164
    },
    {
      "path": "root.state.row_timestep_plan[0][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 1.0,
      "mean": 0.2496371567249298
    },
    {
      "path": "root.state.row_timestep_plan[1][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0017331242561340332,
      "max": 0.9990000128746033,
      "mean": 0.33587655425071716
    },
    {
      "path": "root.state.row_timestep_plan[1][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[2][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0035335421562194824,
      "max": 0.9990000128746033,
      "mean": 0.3388398587703705
    },
    {
      "path": "root.state.row_timestep_plan[2][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[3][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.00540536642074585,
      "max": 0.9990000128746033,
      "mean": 0.3418940007686615
    },
    {
      "path": "root.state.row_timestep_plan[3][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[4][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.007352948188781738,
      "max": 0.9990000128746033,
      "mean": 0.3450433313846588
    },
    {
      "path": "root.state.row_timestep_plan[4][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[5][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.009380877017974854,
      "max": 0.9990000128746033,
      "mean": 0.34829244017601013
    },
    {
      "path": "root.state.row_timestep_plan[5][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[6][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.011494219303131104,
      "max": 0.9990000128746033,
      "mean": 0.3516462743282318
    },
    {
      "path": "root.state.row_timestep_plan[6][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[7][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.013698697090148926,
      "max": 0.9990000128746033,
      "mean": 0.3551100790500641
    },
    {
      "path": "root.state.row_timestep_plan[7][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[8][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.015999972820281982,
      "max": 0.9990000128746033,
      "mean": 0.3586895763874054
    },
    {
      "path": "root.state.row_timestep_plan[8][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[9][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.01840496063232422,
      "max": 0.9990000128746033,
      "mean": 0.3623908460140228
    },
    {
      "path": "root.state.row_timestep_plan[9][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[10][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.02092057466506958,
      "max": 0.9990000128746033,
      "mean": 0.36622026562690735
    },
    {
      "path": "root.state.row_timestep_plan[10][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[11][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.023554623126983643,
      "max": 0.9990000128746033,
      "mean": 0.37018489837646484
    },
    {
      "path": "root.state.row_timestep_plan[11][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[12][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.02631586790084839,
      "max": 0.9990000128746033,
      "mean": 0.3742922842502594
    },
    {
      "path": "root.state.row_timestep_plan[12][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[13][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.029213547706604004,
      "max": 0.9990000128746033,
      "mean": 0.37855052947998047
    },
    {
      "path": "root.state.row_timestep_plan[13][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[14][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.03225815296173096,
      "max": 0.9990000128746033,
      "mean": 0.3829684257507324
    },
    {
      "path": "root.state.row_timestep_plan[14][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[15][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.03546100854873657,
      "max": 0.9990000128746033,
      "mean": 0.38755539059638977
    },
    {
      "path": "root.state.row_timestep_plan[15][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[16][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.03883492946624756,
      "max": 0.9990000128746033,
      "mean": 0.3923218250274658
    },
    {
      "path": "root.state.row_timestep_plan[16][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[17][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.042394042015075684,
      "max": 0.9990000128746033,
      "mean": 0.3972788155078888
    },
    {
      "path": "root.state.row_timestep_plan[17][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[18][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.04615384340286255,
      "max": 0.9990000128746033,
      "mean": 0.4024386405944824
    },
    {
      "path": "root.state.row_timestep_plan[18][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[19][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.050131916999816895,
      "max": 0.9990000128746033,
      "mean": 0.4078145921230316
    },
    {
      "path": "root.state.row_timestep_plan[19][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[20][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.05434781312942505,
      "max": 0.9990000128746033,
      "mean": 0.41342124342918396
    },
    {
      "path": "root.state.row_timestep_plan[20][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[21][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.05882352590560913,
      "max": 0.9990000128746033,
      "mean": 0.4192744791507721
    },
    {
      "path": "root.state.row_timestep_plan[21][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[22][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.06358379125595093,
      "max": 0.9990000128746033,
      "mean": 0.4253920018672943
    },
    {
      "path": "root.state.row_timestep_plan[22][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[23][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.06865674257278442,
      "max": 0.9990000128746033,
      "mean": 0.4317931830883026
    },
    {
      "path": "root.state.row_timestep_plan[23][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[24][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.07407402992248535,
      "max": 0.9990000128746033,
      "mean": 0.43849945068359375
    },
    {
      "path": "root.state.row_timestep_plan[24][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[25][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0798722505569458,
      "max": 0.9990000128746033,
      "mean": 0.44553473591804504
    },
    {
      "path": "root.state.row_timestep_plan[25][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[26][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.08609277009963989,
      "max": 0.9990000128746033,
      "mean": 0.4529256820678711
    },
    {
      "path": "root.state.row_timestep_plan[26][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[27][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.09278357028961182,
      "max": 0.9990000128746033,
      "mean": 0.4607020616531372
    },
    {
      "path": "root.state.row_timestep_plan[27][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[28][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.10000002384185791,
      "max": 0.9990000128746033,
      "mean": 0.4688974618911743
    },
    {
      "path": "root.state.row_timestep_plan[28][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[29][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.1078067421913147,
      "max": 0.9990000128746033,
      "mean": 0.47754982113838196
    },
    {
      "path": "root.state.row_timestep_plan[29][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[30][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.11627912521362305,
      "max": 0.9990000128746033,
      "mean": 0.4867022931575775
    },
    {
      "path": "root.state.row_timestep_plan[30][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[31][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.12550610303878784,
      "max": 0.9990000128746033,
      "mean": 0.4964040219783783
    },
    {
      "path": "root.state.row_timestep_plan[31][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[32][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.13559317588806152,
      "max": 0.9990000128746033,
      "mean": 0.5067117810249329
    },
    {
      "path": "root.state.row_timestep_plan[32][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[33][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.14666670560836792,
      "max": 0.9990000128746033,
      "mean": 0.5176913738250732
    },
    {
      "path": "root.state.row_timestep_plan[33][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[34][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.15887844562530518,
      "max": 0.9990000128746033,
      "mean": 0.5294193625450134
    },
    {
      "path": "root.state.row_timestep_plan[34][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[35][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.17241382598876953,
      "max": 0.9990000128746033,
      "mean": 0.5419864654541016
    },
    {
      "path": "root.state.row_timestep_plan[35][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[36][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.18750005960464478,
      "max": 0.9990000128746033,
      "mean": 0.5555000305175781
    },
    {
      "path": "root.state.row_timestep_plan[36][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[37][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.20441991090774536,
      "max": 0.9990000128746033,
      "mean": 0.5700897574424744
    },
    {
      "path": "root.state.row_timestep_plan[37][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[38][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.22352945804595947,
      "max": 0.9990000128746033,
      "mean": 0.5859135985374451
    },
    {
      "path": "root.state.row_timestep_plan[38][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[39][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.2452830672264099,
      "max": 0.9990000128746033,
      "mean": 0.6031668186187744
    },
    {
      "path": "root.state.row_timestep_plan[39][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[40][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.27027028799057007,
      "max": 0.9990000128746033,
      "mean": 0.6220951080322266
    },
    {
      "path": "root.state.row_timestep_plan[40][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[41][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.2992700934410095,
      "max": 0.9990000128746033,
      "mean": 0.6430131196975708
    },
    {
      "path": "root.state.row_timestep_plan[41][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[42][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.3333333730697632,
      "max": 0.9990000128746033,
      "mean": 0.6663333773612976
    },
    {
      "path": "root.state.row_timestep_plan[42][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[43][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.373913049697876,
      "max": 0.9990000128746033,
      "mean": 0.6926103234291077
    },
    {
      "path": "root.state.row_timestep_plan[43][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[44][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.4230769872665405,
      "max": 0.9990000128746033,
      "mean": 0.7226132750511169
    },
    {
      "path": "root.state.row_timestep_plan[44][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[45][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.4838709831237793,
      "max": 0.9990000128746033,
      "mean": 0.7574481964111328
    },
    {
      "path": "root.state.row_timestep_plan[45][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[46][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.5609756112098694,
      "max": 0.9990000128746033,
      "mean": 0.7987797260284424
    },
    {
      "path": "root.state.row_timestep_plan[46][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[47][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.6619718074798584,
      "max": 0.9990000128746033,
      "mean": 0.8492547869682312
    },
    {
      "path": "root.state.row_timestep_plan[47][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.row_timestep_plan[48][0]",
      "shape": [
        3
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.800000011920929,
      "max": 0.9990000128746033,
      "mean": 0.9133921265602112
    },
    {
      "path": "root.state.row_timestep_plan[48][1]",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.5180515050888062
    },
    {
      "path": "root.state.token_tags",
      "shape": [
        22048
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 2.0,
      "mean": 0.041137516498565674
    },
    {
      "path": "root.state.position_ids",
      "shape": [
        22048,
        3
      ],
      "dtype": "torch.float64",
      "finite": true,
      "min": -2.547236919403076,
      "max": 5792.0,
      "mean": 1650.107666015625
    },
    {
      "path": "root.state.video_indices",
      "shape": [
        16049
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 5585.0,
      "max": 22047.0,
      "mean": 13881.01953125
    },
    {
      "path": "root.state.audio_indices",
      "shape": [
        414
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 11089.0,
      "max": 11502.0,
      "mean": 11295.5
    },
    {
      "path": "root.state.text_indices",
      "shape": [
        5585
      ],
      "dtype": "torch.int64",
      "finite": true,
      "min": 0.0,
      "max": 5584.0,
      "mean": 2792.0
    },
    {
      "path": "root.state.audio_timesteps",
      "shape": [
        49
      ],
      "dtype": "torch.float32",
      "finite": true,
      "min": 0.0,
      "max": 0.9411764740943909,
      "mean": 0.3138476312160492
    }
  ]
}