Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 03:59:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix a typo

Browse Source
This commit is contained in:
行嗔
2022-10-25 00:52:35 +08:00
parent 46c3bdcfe8
commit 85a7832d57
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/trainers/test_ofa_trainer.py
View File

@@ -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',