config.json

{
  "model_type": "spectrum_jeff",
  "output_dims": 2,
  "embed_dim": 384,
  "num_heads": 8,
  "mlp_ratio": 4,
  "patch_size": 32,
  "dropout": 0.1,
  "bottleneck_dim": 6,
  "num_encoder_blocks": 6,
  "num_decoder_blocks": 2,
  "min_period": 0.001,
  "max_period": 20000.0,
  "clip_ivar": 10000,
  "encoder_mask_fraction": 0.0
}