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