From f93c2aec3e1d4d24cece69bd709b8ad4e55019e1 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Wed, 25 Dec 2019 20:37:23 -0400 Subject: [PATCH] next --- cyclegan/models/test_model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py index 65145f5..d7eaa8d 100644 --- a/cyclegan/models/test_model.py +++ b/cyclegan/models/test_model.py @@ -94,6 +94,10 @@ class TestModel(BaseModel): l2 = F.mse_loss(generated, generated_noattack) l0 = (generated - generated_noattack).norm(0) d = (generated - generated_noattack).norm(float('-inf')) + + generated = None + generated_noattack = None + return l1, l2, l0, d def optimize_parameters(self):