try gtp vit

2024-04-28 15:25:44 +07:00
parent 6fc56b920f
commit 08a3780ba6
1 changed files with 2 additions and 2 deletions
--- 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