try gtp vit

This commit is contained in:
thanhvc3 2024-04-28 15:25:44 +07:00
parent 6fc56b920f
commit 08a3780ba6

View File

@ -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