diff --git a/pix2pixHD_attack/test.py b/pix2pixHD_attack/test.py index 0499f0b..d9f208c 100755 --- a/pix2pixHD_attack/test.py +++ b/pix2pixHD_attack/test.py @@ -36,7 +36,7 @@ else: from run_engine import run_trt_engine, run_onnx # Initialize Metrics -l1_error, l2_error, min_dist, l0_error = 0.0 +l1_error, l2_error, min_dist, l0_error = 0.0, 0.0, 0.0, 0.0 n_samples = 0 for i, data in enumerate(dataset):