From 85a7832d575a3ade8210ff9a434df71862dbf16b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A1=8C=E5=97=94?= Date: Tue, 25 Oct 2022 00:52:35 +0800 Subject: [PATCH] fix a typo --- tests/trainers/test_ofa_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/trainers/test_ofa_trainer.py b/tests/trainers/test_ofa_trainer.py index 894e67d2..786599bb 100644 --- a/tests/trainers/test_ofa_trainer.py +++ b/tests/trainers/test_ofa_trainer.py @@ -39,7 +39,7 @@ class TestOfaTrainer(unittest.TestCase): 'dataloader': {'batch_size_per_gpu': 1, 'workers_per_gpu': 0}, 'lr_scheduler': {'name': 'polynomial_decay', 'warmup_proportion': 0.01, - 'lr_endo': 1e-07}, + 'lr_end': 1e-07}, 'lr_scheduler_hook': {'type': 'LrSchedulerHook', 'by_epoch': False}, 'optimizer': {'type': 'AdamW', 'lr': 5e-05, 'weight_decay': 0.01}, 'optimizer_hook': {'type': 'TorchAMPOptimizerHook',