From 22c54c3b72213a99aa21eb98e47cc3b8f7ff0a7f Mon Sep 17 00:00:00 2001 From: RealDream21 Date: Mon, 31 Mar 2025 16:20:09 +0300 Subject: [PATCH] [FIX] Fix typo in clip_models.py --- models/clip_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/clip_models.py b/models/clip_models.py index 44de4db..2dda663 100755 --- a/models/clip_models.py +++ b/models/clip_models.py @@ -12,7 +12,7 @@ class CLIPModel(nn.Module): def __init__(self, name, num_classes=1): super(CLIPModel, self).__init__() - self.model, self.preprocess = clip.load(name, device="cpu") # self.preprecess will not be used during training, which is handled in Dataset class + self.model, self.preprocess = clip.load(name, device="cpu") # self.preprocess will not be used during training, which is handled in Dataset class self.fc = nn.Linear( CHANNELS[name], num_classes )