update

felixbur · Nov 16, 2023 · 5a1fdb4 · 5a1fdb4
1 parent fdc6a9d
commit 5a1fdb4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nkululeko/models/model_cnn.py b/nkululeko/models/model_cnn.py
@@ -202,9 +202,7 @@ def load(self, run, epoch):
         drop = self.util.config_val("MODEL", "drop", False)
         if drop:
             self.util.debug(f"loading: dropout set to: {drop}")
-        self.model = self.MLP(
-            self.feats_train.shape[1], layers, self.class_num, drop
-        ).to(self.device)
+        self.model = myCNN(layers, self.class_num).to(self.device)
         self.model.load_state_dict(torch.load(self.store_path))
         self.model.eval()