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