Merge pull request #30 from QIN2DIM/main

fix(bug): Resolve the conflict
CaptchaAgent · Sep 12, 2022 · bb5a765 · bb5a765
2 parents 7335fd0 + bbeff8d
commit bb5a765
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/factories/resnet.py b/src/factories/resnet.py
@@ -240,6 +240,11 @@ def _train(
 
     def _val(self, model: nn.modules, data_loader: DataLoader):
         total_acc = 0
+        total_tp = 0
+        total_tn = 0
+        total_fp = 0
+        total_fn = 0
+
         for _, (img, label) in enumerate(data_loader):
             img = img.to(self.DEVICE)
             label = label.to(self.DEVICE)