cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add grid search

Browse Source
This commit is contained in:
thanhvc3 2023-05-17 13:45:46 +07:00
parent d912e0a225
commit ff855256e0
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
models.py
View File

@ -537,6 +537,12 @@ class FouriER(torch.nn.Module):
loss = self.bceloss(pred, true_label) loss = self.bceloss(pred, true_label)
return loss 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): def forward_embeddings(self, x):
x = self.patch_embed(x) x = self.patch_embed(x)
return x return x