SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Format

Browse Source
This commit is contained in:
Lysandre
2020-08-27 18:31:51 +02:00
parent 70fccc5cf3
commit 42fddacd1c
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_optimization.py
View File

@@ -154,7 +154,10 @@ class ScheduleInitTest(unittest.TestCase):
self.assertEqual(len([scheduler.get_lr()[0]]), 1) self.assertEqual(len([scheduler.get_lr()[0]]), 1)
lrs_1 = unwrap_schedule(scheduler, self.num_steps) lrs_1 = unwrap_schedule(scheduler, self.num_steps)
self.assertListAlmostEqual( self.assertListAlmostEqual(
lrs_1, expected_learning_rates, tol=1e-2, msg=f"failed for {scheduler_func} in normal scheduler", lrs_1,
expected_learning_rates,
tol=1e-2,
msg=f"failed for {scheduler_func} in normal scheduler",
) )
scheduler = scheduler_func(self.optimizer, **kwargs) scheduler = scheduler_func(self.optimizer, **kwargs)