From 014eb92e04840186dae7a5200b0f72fbc87e5dd7 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 12:25:36 +0100
Subject: [PATCH] typo

---
 models/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/model.py b/models/model.py
index 5853484..1921203 100644
--- a/models/model.py
+++ b/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)
 
 
-- 
GitLab