write all data on training to training log
This commit is contained in:
@@ -176,7 +176,6 @@ def training_entry():
|
||||
loss_fn.backward()
|
||||
opt.step()
|
||||
|
||||
if step % TRAIN_PROGRESS == 0:
|
||||
with torch.no_grad():
|
||||
pred = (torch.sigmoid(logits) > 0.5).float()
|
||||
acc = (pred == y).float().mean().item()
|
||||
|
||||
Reference in New Issue
Block a user