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

changed lbfgs test

Browse Source
This commit is contained in:
William Falcon committed 2019-10-18 01:31:45 +02:00
1 parent d8920169ac
commit 65a2cf6104
1 file changed
+1 -2
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -2
View File
@@ -130,8 +130,7 @@ def test_lbfgs_cpu_model():
reset_seed()
trainer_options = dict(
max_nb_epochs=1,
gradient_clip_val=1.0,
max_nb_epochs=2,
print_nan_grads=True,
show_progress_bar=False,
weights_summary='top',