From 02ba1fe362c48a95ec555e134fe18f6ccb51f27f Mon Sep 17 00:00:00 2001 From: Nicki Skafte Date: Wed, 13 May 2020 17:34:28 +0200 Subject: [PATCH] fix tests --- tests/trainer/test_lr_finder.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/tests/trainer/test_lr_finder.py b/tests/trainer/test_lr_finder.py index 1a7393883e887..b0f6b6d3a0829 100755 --- a/tests/trainer/test_lr_finder.py +++ b/tests/trainer/test_lr_finder.py @@ -174,11 +174,8 @@ def test_suggestion_parameters_work(tmpdir): ) lrfinder = trainer.lr_find(model) - lr1 = lrfinder.suggestion() - lr2 = lrfinder.suggestion(skip_begin=0) - lr3 = lrfinder.suggestion(skip_begin=80) # way too high, should have an impact + lr1 = lrfinder.suggestion(skip_begin=10) # default + lr2 = lrfinder.suggestion(skip_begin=80) # way too high, should have an impact - assert lr1 == lr2, \ - 'Default skipping parameter should not influence suggested learning rate' - assert lr1 != lr3, \ + assert lr1 != lr2, \ 'Skipping parameter did not influence learning rate'