add grid search

2023-05-17 13:45:46 +07:00 · 2023-05-17 13:45:46 +07:00 · ff855256e0
commit ff855256e0
parent d912e0a225
1 changed files with 6 additions and 0 deletions
--- a/models.py
+++ b/models.py
@ -537,6 +537,12 @@ class FouriER(torch.nn.Module):
        loss = self.bceloss(pred, true_label)
        return loss
    
+    def score(self, pred, true_label=None, sub_samp=None):
+        label_pos = true_label[0]
+        label_neg = true_label[1:]
+        loss = self.bceloss(pred, true_label)
+        return loss
+    
    def forward_embeddings(self, x):
        x = self.patch_embed(x)
        return x