wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-27 13:12:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

added safeguards for callbacks in loading saving

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 11:35:55 -04:00
1 parent 2e30dd94bc
commit 3fc8166f51
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
tests/debug.py
+1 -1
View File
@@ -96,7 +96,7 @@ def main():
x, y = batch
x = x.view(x.size(0), -1)
y_hat = model(x)
y_hat = trained_model(x)
# acc
labels_hat = torch.argmax(y_hat, dim=1)