Added the file Architectures

model.py

@@ -243,6 +243,6 @@ def resnet50(pretrained=False, stride=None, num_classes=200, **kwargs):
     if pretrained:
         model.load_state_dict(model_zoo.load_url(
             model_urls['resnet50']), strict=True)
-    if num_classes != 1000:
+    if num_classes != 200:
         model.fc = nn.Linear(512 * Bottleneck.expansion, num_classes)
     return model