From 0be3e5547edbb672823b09a79bd51567846d6711 Mon Sep 17 00:00:00 2001 From: Alex Moreo Date: Sun, 3 May 2020 23:08:39 +0200 Subject: [PATCH] again --- src/model/classifiers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model/classifiers.py b/src/model/classifiers.py index a57defe..6ee413e 100644 --- a/src/model/classifiers.py +++ b/src/model/classifiers.py @@ -46,7 +46,7 @@ class AuthorshipAttributionClassifier(nn.Module): ideal_kernel = torch.as_tensor(1 * (np.outer(1 + yi, 1 / (yi + 1)) == 1)).to(self.device) loss_sav = KernelAlignmentLoss(kernel, ideal_kernel) - loss = loss_attr + loss_sav + loss = loss_attr + 0*loss_sav loss.backward() optim.step()