Thomas Chardonnens commited on
Commit
4d389e0
1 Parent(s): a4a31bd

fix rename

Browse files
Files changed (1) hide show
  1. 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 SeizureDetectionCNN(nn.Module):
7
  def __init__(self, num_classes=2):
8
- super(SeizureDetectionCNN, 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
 
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