Remove test_atk.py try, let the program visually expose the problem

Harry24k · Jan 31, 2024 · bac7731 · bac7731
1 parent 449b0d5
commit bac7731
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/code_coverage/script/pickle_cifar10.py b/code_coverage/script/pickle_cifar10.py
@@ -6,16 +6,16 @@
 ])
 
 testset = datasets.CIFAR10(
-    root='./data', train=False, download=True, transform=transform_test)
+    root='../data', train=False, download=True, transform=transform_test)
 
 test_loader = torch.utils.data.DataLoader(
-    testset, batch_size=100, shuffle=False, num_workers=32)
+    testset, batch_size=10, shuffle=False, num_workers=32)
 
 
 def split(testloader):
     for (x, y) in testloader:
-        torch.save(x, 'images.pt')
-        torch.save(y, 'labels.pt')
+        torch.save(x, 'images.pth')
+        torch.save(y, 'labels.pth')
         break
 
 

diff --git a/code_coverage/script/train_simple_resnet18.py b/code_coverage/script/train_simple_resnet18.py
@@ -7,7 +7,7 @@
 import time
 import os
 
-from resnet import ResNet18
+from code_coverage.script.resnet import ResNet18
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 cudnn.benchmark = True