@ARCH_REGISTRY.register()
class CFSR(nn.Module):
def init(self, in_chans=3, embed_dim=96, depths=(6, 6, 6, 6), dw_size=3,
mlp_ratio=4., upscale=2, img_range=1., upsampler=''):
network structures
network_g:
type: CFSR
upscale: 4
in_chans: 3
dw_size: 9
small_kernel: 3
img_range: 1.
depths: [6, 6]
embed_dim: 48
mlp_ratio: 2
upsampler: 'pixelshuffledirect'
对比可看出,small_kernel这个参数不在CFSR类的参数中
@ARCH_REGISTRY.register()
class CFSR(nn.Module):
def init(self, in_chans=3, embed_dim=96, depths=(6, 6, 6, 6), dw_size=3,
mlp_ratio=4., upscale=2, img_range=1., upsampler=''):
network structures
network_g:
type: CFSR
upscale: 4
in_chans: 3
dw_size: 9
small_kernel: 3
img_range: 1.
depths: [6, 6]
embed_dim: 48
mlp_ratio: 2
upsampler: 'pixelshuffledirect'
对比可看出,small_kernel这个参数不在CFSR类的参数中