try gtp vit

2024-04-28 15:17:40 +07:00
parent d9209a7ef1
commit fddea4769f
1 changed files with 2 additions and 2 deletions
--- a/models.py
+++ b/models.py
@@ -596,9 +596,9 @@ class FouriER(torch.nn.Module):
            graph = degree @ graph @ degree

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