config.json

{
  "Vth": 0.2,
  "a": 0.25,
  "aa": 0.5,
  "architectures": [
    "FSAEModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_fsae.FSAEConfig",
    "AutoModel": "modeling_fsae.FSAEModel"
  },
  "dt": 5,
  "hidden_dims": [
    32,
    64,
    128,
    256
  ],
  "in_channels": 1,
  "k": 20,
  "latent_dim": 128,
  "model_type": "fsae",
  "n_steps": 16,
  "scheduled": true,
  "tau": 0.25,
  "torch_dtype": "float32",
  "transformers_version": "4.35.0"
}