Skip to content

cfsr_arch.py的接口参数与test_CFSR_SRx4.yml中的network structure不一致 #10

Description

@wdddddd

@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类的参数中

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions