v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-19 08:40:34 -07:00
Code Issues Projects Releases Wiki Activity

code convention

Browse Source
This commit is contained in:
Xintao 2021-05-18 21:02:08 +08:00
parent b985478038
commit f565932fb0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
archs/gfpganv1_arch.py
View File

@ -310,7 +310,7 @@ class GFPGANv1(nn.Module):
self.stylegan_decoder.load_state_dict( self.stylegan_decoder.load_state_dict(
torch.load(decoder_load_path, map_location=lambda storage, loc: storage)['params_ema']) torch.load(decoder_load_path, map_location=lambda storage, loc: storage)['params_ema'])
if fix_decoder: if fix_decoder:
for name, param in self.stylegan_decoder.named_parameters(): for _, param in self.stylegan_decoder.named_parameters():
param.requires_grad = False param.requires_grad = False
# for SFT # for SFT