Evanjaa commited on
Commit
5ef3075
1 Parent(s): f2a612b

Added model + weights

Browse files
Files changed (2) hide show
  1. model.py +17 -0
  2. simple_model_weights.pt +3 -0
model.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch.nn as nn
2
+ import torch.nn.functional as F
3
+
4
+ class MultiLayerClassifier(nn.Module):
5
+ def __init__(self, input_size, num_classes):
6
+ super(MultiLayerClassifier, self).__init__()
7
+
8
+ self.fc1 = nn.Linear(input_size, 128, bias=True)
9
+ self.dropout1 = nn.Dropout(0.5)
10
+ self.fc2 = nn.Linear(128, num_classes, bias=True)
11
+
12
+ def forward(self, x):
13
+ x = F.relu(self.fc1(x))
14
+ x = self.dropout1(x)
15
+ x = self.fc2(x)
16
+
17
+ return x
simple_model_weights.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff3063b640c6848aa2b20e09b0f223a63cc4a456d9a096850a5a70b611b83dda
3
+ size 528568