fields     = [temperature specific_humidity velocity_u velocity_v velocity_z]
model_id   = id32n11jh6
token_size = [3, 9, 9]
num_tokens = [12, 6, 12]
ens_dim    = 16
mode       = test
epoch      = 16

velocity_z id32n11jh6 [3, 9, 9] [12, 6, 12] 16