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

update: format and standards

Browse Source
This commit is contained in:
Xintao 2021-09-08 11:28:21 +08:00
parent ad70ce4653
commit f6d3f70646
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gfpgan/archs/gfpganv1_clean_arch.py
View File

@ -230,7 +230,7 @@ class GFPGANv1Clean(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