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

try gtp vit

Browse Source
This commit is contained in:
thanhvc3 2024-04-28 15:17:40 +07:00
parent d9209a7ef1
commit fddea4769f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models.py
View File

@ -596,9 +596,9 @@ class FouriER(torch.nn.Module):
graph = degree @ graph @ degree graph = degree @ graph @ degree
for idx, block in enumerate(self.network): for idx, block in enumerate(self.network):
if (isinstance(block, PoolFormerBlock)): try:
x = block(x, graph) x = block(x, graph)
else: except:
x = block(x) x = block(x)
# output only the features of last layer for image classification # output only the features of last layer for image classification
return x return x