{
  "model_class": "physioex.models.sleeptransformer:SleepTransformer",
  "model_kwargs": {
    "n_classes": 5,
    "in_chan": 1,
    "d_model": 128,
    "n_heads": 8,
    "n_epoch_layers": 4,
    "n_seq_layers": 4,
    "d_ff": 1024,
    "d_clf": 1024,
    "dropout": 0.1,
    "attention_size": 128
  },
  "training": {
    "dataset": "shhs",
    "dataset_kwargs": {
      "visit": 1
    },
    "channels": [
      "EEG"
    ],
    "pipeline_preset": "seqsleepnet",
    "sequence_length": 21,
    "max_epochs": 50,
    "lr": 0.0001,
    "weight_decay": 0,
    "batch_size": 32,
    "loss": "CrossEntropyLoss",
    "fold": 0,
    "early_stopping_patience": 10
  },
  "reference": "Phan et al. 2022 - SleepTransformer: Automatic Sleep Staging with Interpretability and Uncertainty Quantification (arXiv:2105.11043)"
}