{
  "feat_key": "ssl_feat",
  "mag_key": "mag",
  "num_layers": 12,
  "input_channels": 1024,
  "hidden_channels": 512,
  "vit_mlp_dim": 2048,
  "p_uncond": 0.1
}