Method__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
basicsr/models/archs/layers.py:8
Method__init__(self, in_dim=3, out_dim=3, dim=31, level=2, num_blocks=[2, 4, 4])
basicsr/models/archs/RetinexFormer_arch.py:234
Method__init__(self, in_channels=3, out_channels=3, n_feat=31, level=2, num_blocks=[1, 1, 1])
basicsr/models/archs/RetinexFormer_arch.py:325
Method__init__(self, in_channels=3, out_channels=3, n_feat=31, stage=3, num_blocks=[1,1,1])
basicsr/models/archs/RetinexFormer_arch.py:344
Functionconv(in_channels, out_channels, kernel_size, bias=False, padding = 1, stride = 1)
basicsr/models/archs/MST_Plus_Plus_arch.py:50
Methodforward x_in: [b,h,w,c] # input_feature illu_fea: [b,h,w,c] # mask shift? 为什么是 b, h, w, c? return out: [b,h,w,c]
basicsr/models/archs/RetinexFormer_arch.py:147
Methodforward x: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/RetinexFormer_arch.py:194
Methodforward x: [b,c,h,w] illu_fea: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:219
Methodforward x: [b,c,h,w] x是feature, 不是image illu_fea: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:287
Methodforward x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/RetinexFormer_arch.py:353
Methodforward x_in: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/MST_Plus_Plus_arch.py:88
Methodforward x: [b,h,w,c] return out: [b,h,w,c]
basicsr/models/archs/MST_Plus_Plus_arch.py:130
Methodforward x: [b,c,h,w] return out: [b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:154
Methodforward x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:218
Methodforward x: [b,c,h,w] return out:[b,c,h,w]
basicsr/models/archs/MST_Plus_Plus_arch.py:257
Methodforward Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
basicsr/models/losses/losses.py:44
Methodforward Args: pred (Tensor): of shape (N, C, H, W). Predicted tensor. target (Tensor): of shape (N, C, H, W). Ground truth te
basicsr/models/losses/losses.py:73