123456/roma_unsb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove netF and rename netG

Browse Source 
bug
This commit is contained in:
123456 2025-02-23 15:51:57 +08:00
parent 55335dff16
commit 45d12a10d6
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
models/roma_unsb_model.py
View File

@ -253,12 +253,11 @@ class RomaUnsbModel(BaseModel):
self.visual_names += ['idt_B'] self.visual_names += ['idt_B']
if self.isTrain: if self.isTrain:
self.model_names = ['G1', 'F1', 'D1', 'E1', self.model_names = ['G', 'D', 'E']
'G2']
else: 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) 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)