try gtp vit

This commit is contained in:
thanhvc3 2024-04-28 12:05:57 +07:00
parent 68a94bd1e2
commit 541c4fa2b3

View File

@ -541,7 +541,7 @@ class FouriER(torch.nn.Module):
row2, col2 = col_indices // int(math.sqrt(N)), col_indices % int(math.sqrt(N))
graph = ((abs(row1 - row2) <= 1).float() * (abs(col1 - col2) <= 1).float())
graph = graph - torch.eye(N)
self.spatial_graph = graph.to("cuda") # comment .to("cuda") if the environment is cpu
self.spatial_graph = graph.cuda() # comment .to("cuda") if the environment is cpu
self.class_token = False
self.token_scale = False
self.head = nn.Linear(