From 737340d63fb772e641726c6f73bba0fad5cc6af2 Mon Sep 17 00:00:00 2001 From: Lukas Wutschitz Date: Fri, 23 Feb 2024 10:59:51 +0000 Subject: [PATCH] Fix dataloader --- .../predict_with_cnn_classifier.py | 11 +++++------ privacy_estimates/experiments/attacks/dpd/attack.py | 4 ++++ 2 files changed, 9 insertions(+), 6 deletions(-) diff --git a/experiments/components/predict-with-cnn-classifier/predict_with_cnn_classifier.py b/experiments/components/predict-with-cnn-classifier/predict_with_cnn_classifier.py index cfb2b35..3fcb697 100644 --- a/experiments/components/predict-with-cnn-classifier/predict_with_cnn_classifier.py +++ b/experiments/components/predict-with-cnn-classifier/predict_with_cnn_classifier.py @@ -39,14 +39,13 @@ def main(args: Arguments): print(f"Loaded dataset: {data.features}") - if len(data) == 0: - dataset = TensorDataset(torch.tensor([]), torch.tensor([])) - else: - dataset = TensorDataset(torch.tensor(data["image"]), torch.tensor(data["label"])) - model = CNN.load(args.experiment_dir / args.model_rel_path / "model.pt") - data_loader = DataLoader(dataset, batch_size=args.batch_size, shuffle=False) + data_loader = DataLoader( + data.with_format("torch"), + batch_size=args.batch_size, + shuffle=False + ) device = "cpu" if args.use_cpu else "cuda" diff --git a/privacy_estimates/experiments/attacks/dpd/attack.py b/privacy_estimates/experiments/attacks/dpd/attack.py index a13d212..0b5358c 100644 --- a/privacy_estimates/experiments/attacks/dpd/attack.py +++ b/privacy_estimates/experiments/attacks/dpd/attack.py @@ -1,3 +1,7 @@ +""" Implements the white-box attack of the paper "Tight Auditing of Differentially Private Machine Learning" + +[1] Nasr, Milad, Jamie Hayes, Thomas Steinke, Borja Balle, Florian Tramèr, Matthew Jagielski, Nicholas Carlini, and Andreas Terzis. "Tight Auditing of Differentially Private Machine Learning." arXiv preprint arXiv:2302.07956 (2023). +""" import torch import numpy as np from torch.optim import Optimizer