added more models for testing

2026-05-05 21:04:33 +02:00
parent 1c04344ad6
commit 4cc9fa2bac
11 changed files with 143 additions and 21 deletions
--- a/architectures/DenseNet121.py
+++ b/architectures/DenseNet121.py
@@ -12,4 +12,4 @@ class DenseNet121(Model):
        num_ftrs = m.classifier.in_features
        m.classifier = nn.Linear(num_ftrs, self.size)
        
-        return m.to(self.device)
+        return m