try gtp vit

This commit is contained in:
thanhvc3 2024-04-28 15:08:08 +07:00
parent 541c4fa2b3
commit 4daa40527b

View File

@ -596,7 +596,7 @@ 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):
x = block(x) x = block(x, graph)
# output only the features of last layer for image classification # output only the features of last layer for image classification
return x return x