again

2020-05-03 23:08:39 +02:00 · 2020-05-03 23:08:39 +02:00 · 0be3e5547e
parent f118b91ed0
commit 0be3e5547e
1 changed files with 1 additions and 1 deletions
--- 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()