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()