cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try modify swin

Browse Source
This commit is contained in:
thanhvc3 2024-04-29 17:15:54 +07:00
parent 805d4fb536
commit b661823661
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
models.py
View File

@ -565,12 +565,12 @@ class FouriER(torch.nn.Module):
if downsamples[i] or embed_dims[i] != embed_dims[i+1]:
# downsampling between two stages
network.append(
# PatchEmbed(
# patch_size=down_patch_size, stride=down_stride,
# padding=down_pad,
# in_chans=embed_dims[i], embed_dim=embed_dims[i+1]
# )
PatchMerging(dim=embed_dims[i+1])
PatchEmbed(
patch_size=down_patch_size, stride=down_stride,
padding=down_pad,
in_chans=embed_dims[i], embed_dim=embed_dims[i+1]
)
# PatchMerging(dim=embed_dims[i+1])
)
self.network = nn.ModuleList(network)