Spaces:
Runtime error
Runtime error
Commit
·
9dfdffe
1
Parent(s):
98c7fa2
updated
Browse files
app/Hackathon_setup/exp_recognition_model.py
CHANGED
|
@@ -15,7 +15,7 @@ classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5:
|
|
| 15 |
# Example Network
|
| 16 |
class facExpRec(torch.nn.Module):
|
| 17 |
def __init__(self):
|
| 18 |
-
|
| 19 |
self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
|
| 20 |
self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
|
| 21 |
self.conv3 = nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3)
|
|
|
|
| 15 |
# Example Network
|
| 16 |
class facExpRec(torch.nn.Module):
|
| 17 |
def __init__(self):
|
| 18 |
+
super(Model, self).__init__()
|
| 19 |
self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
|
| 20 |
self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
|
| 21 |
self.conv3 = nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3)
|