From ef312eec398ce908a8dbd5fd4e5dab147b4b045d Mon Sep 17 00:00:00 2001 From: Viktor Valadi Date: Mon, 21 Oct 2024 14:55:29 +0000 Subject: [PATCH] bug fix --- leakpro/metrics/attack_result.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/leakpro/metrics/attack_result.py b/leakpro/metrics/attack_result.py index 990d271..1bbaf1c 100755 --- a/leakpro/metrics/attack_result.py +++ b/leakpro/metrics/attack_result.py @@ -169,7 +169,7 @@ def extract_tensors_from_subset(dataset: Dataset) -> Tensor: output_denormalized = clamp(recreated_data * self.data_std + self.data_mean, 0, 1) - os.makedirs(save_path, exist_ok=True) + os.makedirs(save_path + "/" + attack_name, exist_ok=True) save_image(output_denormalized, os.path.join(save_path + "/" + attack_name, "recreated_image.png"))