From 887eda058ae81db913c4cb0c61a15a90050b0ad7 Mon Sep 17 00:00:00 2001 From: KaiGuo20 Date: Mon, 13 Nov 2023 17:48:18 -0500 Subject: [PATCH] Update mettack.py Solution to OOM in Metattack in a higher version of Pytorch --- deeprobust/graph/global_attack/mettack.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deeprobust/graph/global_attack/mettack.py b/deeprobust/graph/global_attack/mettack.py index bb486e5..ad03172 100644 --- a/deeprobust/graph/global_attack/mettack.py +++ b/deeprobust/graph/global_attack/mettack.py @@ -125,9 +125,9 @@ def log_likelihood_constraint(self, modified_adj, ori_adj, ll_cutoff): def get_adj_score(self, adj_grad, modified_adj, ori_adj, ll_constraint, ll_cutoff): adj_meta_grad = adj_grad * (-2 * modified_adj + 1) # Make sure that the minimum entry is 0. - adj_meta_grad -= adj_meta_grad.min() + adj_meta_grad = adj_meta_grad - adj_meta_grad.min() # Filter self-loops - adj_meta_grad -= torch.diag(torch.diag(adj_meta_grad, 0)) + adj_meta_grad = adj_meta_grad - torch.diag(torch.diag(adj_meta_grad, 0)) # # Set entries to 0 that could lead to singleton nodes. singleton_mask = self.filter_potential_singletons(modified_adj) adj_meta_grad = adj_meta_grad * singleton_mask