diff --git a/archs/gfpganv1_arch.py b/archs/gfpganv1_arch.py index a65e603..38a450f 100644 --- a/archs/gfpganv1_arch.py +++ b/archs/gfpganv1_arch.py @@ -310,7 +310,7 @@ class GFPGANv1(nn.Module): self.stylegan_decoder.load_state_dict( torch.load(decoder_load_path, map_location=lambda storage, loc: storage)['params_ema']) if fix_decoder: - for name, param in self.stylegan_decoder.named_parameters(): + for _, param in self.stylegan_decoder.named_parameters(): param.requires_grad = False # for SFT