Upload seqsleepnet-phan/config.json with huggingface_hub

seqsleepnet-phan/config.json

@@ -16,7 +16,14 @@
     "seqnlayer2": 4
   },
   "training": {
-    "dataset": "sleepedf",
+    "dataset": "mass",
+    "dataset_cohorts": [
+      1,
+      2,
+      3,
+      4,
+      5
+    ],
     "channels": [
       "EEG"
     ],
@@ -27,7 +34,8 @@
     "weight_decay": 0,
     "batch_size": 32,
     "loss": "CrossEntropyLoss",
-    "fold": 0
+    "fold": 0,
+    "early_stopping_patience": 10
   },
   "reference": "Phan et al. 2019 - SeqSleepNet: End-to-End Hierarchical Recurrent Neural Network for Sequence-to-Sequence Automatic Sleep Staging"
 }