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