From b985478038aacb07d6d8a2a6981e2866139d82b3 Mon Sep 17 00:00:00 2001 From: Xintao Date: Tue, 18 May 2021 20:56:37 +0800 Subject: [PATCH] code convention --- archs/arcface_arch.py | 2 +- archs/gfpganv1_arch.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/archs/arcface_arch.py b/archs/arcface_arch.py index a2b41be..9411fcf 100644 --- a/archs/arcface_arch.py +++ b/archs/arcface_arch.py @@ -174,7 +174,7 @@ class ResNetArcFace(nn.Module): layers = [] layers.append(block(self.inplanes, planes, stride, downsample, use_se=self.use_se)) self.inplanes = planes - for i in range(1, blocks): + for _ in range(1, blocks): layers.append(block(self.inplanes, planes, use_se=self.use_se)) return nn.Sequential(*layers) diff --git a/archs/gfpganv1_arch.py b/archs/gfpganv1_arch.py index 48c4840..a65e603 100644 --- a/archs/gfpganv1_arch.py +++ b/archs/gfpganv1_arch.py @@ -30,7 +30,7 @@ class StyleGAN2GeneratorSFT(StyleGAN2Generator): num_style_feat=512, num_mlp=8, channel_multiplier=2, - resample_kernel=[1, 3, 3, 1], + resample_kernel=(1, 3, 3, 1), lr_mlp=0.01, narrow=1, sft_half=False): @@ -231,7 +231,7 @@ class GFPGANv1(nn.Module): out_size, num_style_feat=512, channel_multiplier=1, - resample_kernel=[1, 3, 3, 1], + resample_kernel=(1, 3, 3, 1), decoder_load_path=None, fix_decoder=True, # for stylegan decoder