AssertionError: shift_size must in 0-window_size

[alqurri77]

I got below error. Could you please advise where the problem could be.

SwinTransformerSys expand initial----depths:[2, 2, 6, 2];depths_decoder:[1, 2, 2, 2];drop_path_rate:0.1;num_classes:1000

AssertionError Traceback (most recent call last)
C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/3765231964.py in
127 if not os.path.exists(args.output_dir):
128 os.makedirs(args.output_dir)
--> 129 net = SwinUnet(None, img_size=args.img_size, num_classes=args.num_classes).cuda() #ViT_seg(config, img_size=args.img_size, num_classes=args.num_classes).cuda()
130 net.load_from(config)
131

C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/1252163837.py in init(self, config, img_size, num_classes, zero_head, vis)
28 #self.config = config
29
---> 30 self.swin_unet = SwinTransformerSys(img_size=224 ,#config.DATA.IMG_SIZE,
31 patch_size=128,#config.MODEL.SWIN.PATCH_SIZE,
32 in_chans=3,#config.MODEL.SWIN.IN_CHANS,

C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/1477083531.py in init(self, img_size, patch_size, in_chans, num_classes, embed_dim, depths, depths_decoder, num_heads, window_size, mlp_ratio, qkv_bias, qk_scale, drop_rate, attn_drop_rate, drop_path_rate, norm_layer, ape, patch_norm, use_checkpoint, final_upsample, **kwargs)
624 self.layers = nn.ModuleList()
625 for i_layer in range(self.num_layers):
--> 626 layer = BasicLayer(dim=int(embed_dim * 2 ** i_layer),
627 input_resolution=(patches_resolution[0] // (2 ** i_layer),
628 patches_resolution[1] // (2 ** i_layer)),

C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/1477083531.py in init(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio, qkv_bias, qk_scale, drop, attn_drop, drop_path, norm_layer, downsample, use_checkpoint)
412
413 # build blocks
--> 414 self.blocks = nn.ModuleList([
415 SwinTransformerBlock(dim=dim, input_resolution=input_resolution,
416 num_heads=num_heads, window_size=window_size,

C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/1477083531.py in (.0)
413 # build blocks
414 self.blocks = nn.ModuleList([
--> 415 SwinTransformerBlock(dim=dim, input_resolution=input_resolution,
416 num_heads=num_heads, window_size=window_size,
417 shift_size=0 if (i % 2 == 0) else window_size // 2,

C:\Users\60962~1.LPI\AppData\Local\Temp/ipykernel_17660/1477083531.py in init(self, dim, input_resolution, num_heads, window_size, shift_size, mlp_ratio, qkv_bias, qk_scale, drop, attn_drop, drop_path, act_layer, norm_layer)
188 self.shift_size = 0
189 self.window_size = min(self.input_resolution)
--> 190 assert 0 <= self.shift_size < self.window_size, "shift_size must in 0-window_size"
191
192 self.norm1 = norm_layer(dim)

AssertionError: shift_size must in 0-window_size