model_config.json

{
  "embed_dim": 64,
  "hidden_dim": 256,
  "n_heads": 8,
  "encoder_layers": 8,
  "encoder_mlp_dim": 1024,
  "predictor_layers": 4,
  "predictor_mlp_dim": 512,
  "latent_dim": 256,
  "window_size": 256,
  "dropout": 0.0,
  "layer_norm_eps": 1e-05
}