From bac77317066469d62160e3c7e83045937efb5b16 Mon Sep 17 00:00:00 2001 From: Riko Date: Wed, 31 Jan 2024 18:02:32 +0800 Subject: [PATCH] Remove test_atk.py `try`, let the program visually expose the problem --- code_coverage/script/pickle_cifar10.py | 8 ++++---- code_coverage/script/train_simple_resnet18.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/code_coverage/script/pickle_cifar10.py b/code_coverage/script/pickle_cifar10.py index edc2d232..c5bdddd1 100644 --- 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 index 99a4115d..75179c95 100644 --- 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