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

fixed amp bug

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 14:16:05 -04:00
1 parent ca1835e063
commit 5fe833ae01
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -1
View File
@@ -281,7 +281,7 @@ def run_prediction(dataloader, trained_model):
print(val_acc)
assert val_acc > 0.70, f'this model is expected to get > 0.7 in test set (it got {val_acc})'
assert val_acc > 0.60, f'this model is expected to get > 0.6 in test set (it got {val_acc})'
def assert_ok_acc(trainer):