Upload folder using huggingface_hub

README.md

@@ -0,0 +1,3 @@
+# marianaagdias/ecg_peak_detection
+
+

hparams.yaml

@@ -0,0 +1,11 @@
+bidirectional: true
+dropout: 0
+hid_dim:
+- 32
+- 64
+- 64
+learning_rate: 0.001
+n_features: 1
+n_layers: 3
+num_classes: 1
+task: classification

model_weights.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67c5326c142ff01ec3e911f2459b42826392bf06dcc53ca2d26b051daaa785df
+size 533628

training_info.json

@@ -0,0 +1,12 @@
+{
+    "model": "GRUseq2seq",
+    "train_dataset": "private_gib01",
+    "task": "peak detection",
+    "gpu_model": "NVIDIA GeForce GTX 1080 Ti",
+    "epochs": 80,
+    "optimizer": "Adam (\nParameter Group 0\n    amsgrad: False\n    betas: (0.9, 0.999)\n    capturable: False\n    differentiable: False\n    eps: 1e-08\n    foreach: None\n    fused: None\n    initial_lr: 0.001\n    lr: 0.001\n    maximize: False\n    weight_decay: 1e-05\n)",
+    "learning_rate": 0.001,
+    "validation_loss": 0.14879398047924042,
+    "training_time": 11375.492486476898,
+    "retraining": false
+}