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
|