From fddea4769ffa8254dc4dce07a01b5196991eb958 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Sun, 28 Apr 2024 15:17:40 +0700
Subject: [PATCH] try gtp vit

---
 models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models.py b/models.py
index 223a03e..db6be39 100644
--- 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