Base Swin Unet and Trans UNET

[Tombery1]

Hi, Please, I want the closest configuration to the original paper Swin Unet

outputs = base.swin_unet_2d_base(input_img, filter_num_begin=64, depth=4, stack_num_down=2, stack_num_up=2, 
            patch_size=(2, 2), num_heads=[4, 8, 8, 8], window_size=[4, 2, 2, 2], num_mlp=512, name='swin_unet')

Trans UNET

outputs = base.transunet_2d_base(input_img, filter_num=[16, 64, 128,256, 512], name='transunet')