wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove summary

Browse Source
This commit is contained in:
William Falcon
2019-12-11 06:50:44 -08:00
parent 23f1b98c0e
commit 28b68503bc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/utils.py
+1 -1
View File
@@ -159,7 +159,7 @@ def run_prediction(dataloader, trained_model, dp=False, min_acc=0.50):
acc = torch.mean(acc).item()
else:
y_hat = trained_model(x)
y_hat = trained_model(x, 0)
# acc
labels_hat = torch.argmax(y_hat, dim=1)