From 45d12a10d6302e30a96659d7d9e78bec5544fc2f Mon Sep 17 00:00:00 2001
From: 123456 <3351416005@qq.com>
Date: Sun, 23 Feb 2025 15:51:57 +0800
Subject: [PATCH] remove netF and rename netG

bug
---
 models/roma_unsb_model.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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)