This commit is contained in:
Nataniel Ruiz
2019-12-25 15:36:45 -04:00
parent 69da86bd56
commit 4a835bb61f
+1 -1
View File
@@ -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):