change typo

khrlimam · Jul 30, 2019 · 1c7e51c · 1c7e51c
1 parent 58c69c8
commit 1c7e51c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/models.py b/models.py
@@ -15,6 +15,7 @@ def __init__(self, pretrained=False):
 
         self.model = resnet50(pretrained)
         self.embedding_size = 128
+        num_classes = 500
         self.cnn = nn.Sequential(
             self.model.conv1,
             self.model.bn1,
@@ -33,7 +34,7 @@ def __init__(self, pretrained=False):
             # nn.ReLU(),
             nn.Linear(100352, self.embedding_size))
 
-	self.model.classifier = nn.Linear(self.embedding_size, num_classes)
+        self.model.classifier = nn.Linear(self.embedding_size, num_classes)
 
     def l2_norm(self, input):
         input_size = input.size()