try gtp vit

2024-04-28 15:08:08 +07:00
parent 541c4fa2b3
commit 4daa40527b
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -596,7 +596,7 @@ class FouriER(torch.nn.Module):
            graph = degree @ graph @ degree

        for idx, block in enumerate(self.network):
-            x = block(x)
+            x = block(x, graph)
        # output only the features of last layer for image classification
        return x