bug fix

aidotse · Oct 21, 2024 · ef312ee · ef312ee
1 parent a388873
commit ef312ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"))