Open laulampaul opened 1 year ago
Can you provide more information of your running conditions?
python== 3.8.8 pytorch==1.9.0 torchvision==0.9.0
I mean how do you run the psrt_sliding_arch.py? Your input size or other information
just use the code in psrt_sliding_arch.py
if name == 'main': upscale = 4 window_size = (2, 8, 8) height = (1024 // upscale // window_size[1] + 1) window_size[1] width = (1024 // upscale // window_size[2] + 1) window_size[2]
model = SwinIRFM( img_size=height, patch_size=1, in_chans=3, embed_dim=96, depths=(6, 6, 6, 6), num_heads=(6, 6, 6, 6), window_size=window_size, mlp_ratio=2, qkv_bias=True, qk_scale=None, drop_rate=0., attn_drop_rate=0., drop_path_rate=0.1, norm_layer=nn.LayerNorm, ape=False, patch_norm=True, use_checkpoint=False, upscale=2, img_range=1., upsampler='pixelshuffle', ) print(model) #print(height, width, model.flops() / 1e9) x = torch.randn((1, 5, 3, height, width)) x = model(x) print(x.shape)
I also show this problem: TypeError: can only concatenate tuple (not "int") to tuple. How can I solve it?
Meet the same problem. How to solve it?
I also meet the question, what can I do??
When I run the code [psrt_sliding_arch.py] , I face one bug
And I try to change attn_windows = attn_windows.view(-1, *(self.window_size + [c,])) to attn_windows = attn_windows.view(-1, self.window_size[0], self.window_size[1],self.window_size[2],c)
And face a new bug
How can I solve it ??