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

Tests: refactor cleanup (#1744)

Browse Source 
* wip

* cleaning

* optim imports

* -

* default hparams

* fix restore

* fix imports
This commit is contained in:
Jirka Borovec
2020-05-10 13:15:28 -04:00
committed by GitHub
parent 4970927ec8
commit 134eb61e1a
41 changed files with 187 additions and 1150 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/callbacks/test_progress_bar.py
+3 -3
View File
@@ -58,7 +58,7 @@ def test_progress_bar_misconfiguration():
def test_progress_bar_totals():
"""Test that the progress finishes with the correct total steps processed."""
model = EvalModelTemplate(tutils.get_default_hparams())
model = EvalModelTemplate()
trainer = Trainer(
progress_bar_refresh_rate=1,
@@ -107,7 +107,7 @@ def test_progress_bar_totals():
def test_progress_bar_fast_dev_run():
model = EvalModelTemplate(tutils.get_default_hparams())
model = EvalModelTemplate()
trainer = Trainer(
fast_dev_run=True,
@@ -140,7 +140,7 @@ def test_progress_bar_fast_dev_run():
def test_progress_bar_progress_refresh(refresh_rate):
"""Test that the three progress bars get correctly updated when using different refresh rates."""
model = EvalModelTemplate(tutils.get_default_hparams())
model = EvalModelTemplate()
class CurrentProgressBar(ProgressBar):