Thomas Chardonnens
commited on
Commit
•
4d389e0
1
Parent(s):
a4a31bd
fix rename
Browse files- seizure_detection.py +2 -2
seizure_detection.py
CHANGED
@@ -3,9 +3,9 @@ import torch.nn as nn
|
|
3 |
import torch.nn.functional as F
|
4 |
|
5 |
|
6 |
-
class
|
7 |
def __init__(self, num_classes=2):
|
8 |
-
super(
|
9 |
self.conv1= nn.Conv2d(1, 32, kernel_size=3, stride=1, padding=1) # 32, 224, 224
|
10 |
|
11 |
self.pool= nn.MaxPool2d(kernel_size=2, stride=2) # 32, 112, 112
|
|
|
3 |
import torch.nn.functional as F
|
4 |
|
5 |
|
6 |
+
class SeizureDetector(nn.Module):
|
7 |
def __init__(self, num_classes=2):
|
8 |
+
super(SeizureDetector, self).__init__()
|
9 |
self.conv1= nn.Conv2d(1, 32, kernel_size=3, stride=1, padding=1) # 32, 224, 224
|
10 |
|
11 |
self.pool= nn.MaxPool2d(kernel_size=2, stride=2) # 32, 112, 112
|