diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index b4df892..71ae639 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -253,12 +253,11 @@ class RomaUnsbModel(BaseModel):
             self.visual_names += ['idt_B']
 
         if self.isTrain:
-            self.model_names = ['G1', 'F1', 'D1', 'E1',
-                                'G2']
+            self.model_names = ['G', 'D', 'E']
 
             
         else:
-            self.model_names = ['G1']
+            self.model_names = ['G']
             
         # 创建网络
         self.netG = networks.define_G(opt.input_nc, opt.output_nc, opt.ngf, opt.netG, opt.normG, not opt.no_dropout, opt.init_type, opt.init_gain, opt.no_antialias, opt.no_antialias_up, self.gpu_ids, opt)