From 08a3780ba6dd3230cf66da5d77568bef00cc0e9c Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Sun, 28 Apr 2024 15:25:44 +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 ffab680..464c74f 100644
--- a/models.py
+++ b/models.py
@@ -597,7 +597,7 @@ class FouriER(torch.nn.Module):
 
         for idx, block in enumerate(self.network):
             try:
-                x = block((x, graph))
+                x = block(x, graph)
             except:
                 x = block(x)
         # output only the features of last layer for image classification
@@ -758,7 +758,7 @@ def basic_blocks(dim, index, layers,
             use_layer_scale=use_layer_scale, 
             layer_scale_init_value=layer_scale_init_value, 
             ))
-    blocks = SeqModel(*blocks)
+    blocks = SeqModel(blocks)
 
     return blocks