From 1ab84c807cae2010d146898990cd85790913ca6c Mon Sep 17 00:00:00 2001 From: araison Date: Fri, 30 Dec 2022 19:47:57 +0100 Subject: [PATCH] Fixing bugs and adding new features --- explaining_framework/metric/robust.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/explaining_framework/metric/robust.py b/explaining_framework/metric/robust.py index 4e74654..b4348d9 100644 --- a/explaining_framework/metric/robust.py +++ b/explaining_framework/metric/robust.py @@ -2,10 +2,10 @@ import copy import torch import torch.nn.functional as F +from torch.nn import CrossEntropyLoss, MSELoss from torch_geometric.explain.explanation import Explanation from torch_geometric.graphgym.config import cfg from torch_geometric.utils import add_random_edge, dropout_edge, dropout_node -from troch.nn import CrossEntropyLoss, MSELoss from explaining_framework.metric.base import Metric