nivirx/mltoys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

write all data on training to training log

Browse Source
This commit is contained in:
Elaina Claus
2025-12-25 02:07:49 +00:00
parent 921e24b451
commit 4f500e8b4c
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pairwise_compare.py
View File

@@ -176,7 +176,6 @@ def training_entry():
loss_fn.backward() loss_fn.backward()
opt.step() opt.step()
if step % TRAIN_PROGRESS == 0:
with torch.no_grad(): with torch.no_grad():
pred = (torch.sigmoid(logits) > 0.5).float() pred = (torch.sigmoid(logits) > 0.5).float()
acc = (pred == y).float().mean().item() acc = (pred == y).float().mean().item()