try gtp vit
This commit is contained in:
parent
6fc56b920f
commit
08a3780ba6
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user