Spaces:
Sleeping
Sleeping
Commit
·
b27a86b
1
Parent(s):
c1bc8a5
fix
Browse files
app/Hackathon_setup/face_recognition_model.py
CHANGED
|
@@ -11,7 +11,7 @@ from torchvision import transforms
|
|
| 11 |
trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
|
| 12 |
|
| 13 |
##Example Network
|
| 14 |
-
class
|
| 15 |
def __init__(self):
|
| 16 |
super(SiameseNetwork, self).__init__()
|
| 17 |
self.cnn1 = nn.Sequential(
|
|
|
|
| 11 |
trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
|
| 12 |
|
| 13 |
##Example Network
|
| 14 |
+
class SiameseNetwork(torch.nn.Module):
|
| 15 |
def __init__(self):
|
| 16 |
super(SiameseNetwork, self).__init__()
|
| 17 |
self.cnn1 = nn.Sequential(
|