typo

models/model.py

@@ -265,7 +265,7 @@ class Classification_model(nn.Module):
     def __init__(self, model, n_class, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.n_class = n_class
-        if model =='Resnet18':
+        if model =='ResNet18':
             self.im_encoder = resnet18(num_classes=self.n_class)