add grid search

2023-05-17 13:45:46 +07:00
parent d912e0a225
commit ff855256e0
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