Fix logger bug and prepare data bug (#1933)

* tests, fix logger bug and prepare data bug

* add CHANGELOG.md

Co-authored-by: Nicki Skafte <nugginea@gmail.com>
This commit is contained in:
Nicki Skafte
2020-05-25 07:43:56 -04:00
committed by GitHub
co-authored by Nicki Skafte
parent 033ddc0c29
commit a34eb9e169
4 changed files with 54 additions and 1 deletions
+23
View File
@@ -199,3 +199,26 @@ def test_suggestion_with_non_finite_values(tmpdir):
assert before_lr == after_lr, \
'Learning rate was altered because of non-finite loss values'
def test_logger_reset_correctly(tmpdir):
""" Test that logger is updated correctly """
tutils.reset_seed()
hparams = EvalModelTemplate.get_default_hparams()
model = EvalModelTemplate(hparams)
trainer = Trainer(
default_save_path=tmpdir,
max_epochs=10,
auto_lr_find=True
)
logger1 = trainer.logger
trainer.fit(model)
logger2 = trainer.logger
logger3 = model.logger
assert logger1 == logger2, \
'Learning rate finder altered the logger of trainer'
assert logger2 == logger3, \
'Learning rate finder altered the logger of model'
+23
View File
@@ -128,3 +128,26 @@ def test_error_on_dataloader_passed_to_fit(tmpdir):
with pytest.raises(MisconfigurationException):
trainer.fit(model, **fit_options)
def test_logger_reset_correctly(tmpdir):
""" Test that logger is updated correctly """
tutils.reset_seed()
hparams = EvalModelTemplate.get_default_hparams()
model = EvalModelTemplate(hparams)
trainer = Trainer(
default_save_path=tmpdir,
max_epochs=1,
auto_scale_batch_size=True
)
logger1 = trainer.logger
trainer.fit(model)
logger2 = trainer.logger
logger3 = model.logger
assert logger1 == logger2, \
'Batch size finder altered the logger of trainer'
assert logger2 == logger3, \
'Batch size finder altered the logger of model'