From 513b6cde948c08caede2ff66a308c9c85dc55bd7 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Wed, 25 Dec 2019 21:32:52 -0400
Subject: [PATCH] next

---
 cyclegan/models/test_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py
index c71f708..4c3be38 100644
--- a/cyclegan/models/test_model.py
+++ b/cyclegan/models/test_model.py
@@ -42,7 +42,7 @@ class TestModel(BaseModel):
         # specify the training losses you want to print out. The training/test scripts  will call <BaseModel.get_current_losses>
         self.loss_names = []
         # specify the images you want to save/display. The training/test scripts  will call <BaseModel.get_current_visuals>
-        self.visual_names = ['real', 'fake']
+        self.visual_names = ['real', 'fake', 'fake_noattack']
         # specify the models you want to save to the disk. The training/test scripts will call <BaseModel.save_networks> and <BaseModel.load_networks>
         self.model_names = ['G' + opt.model_suffix]  # only generator is needed.
         self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, opt.netG,