From 2863a8f7fa927af48ca72687d748b67ea98a09b6 Mon Sep 17 00:00:00 2001 From: "suluyan.sly" Date: Fri, 2 Dec 2022 17:09:06 +0800 Subject: [PATCH] [to #42322933] fix hook.__init__ Link: https://code.aone.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10957489 * fix hook.__init__ --- modelscope/trainers/hooks/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/trainers/hooks/__init__.py b/modelscope/trainers/hooks/__init__.py index 94a5b613..c7bd93aa 100644 --- a/modelscope/trainers/hooks/__init__.py +++ b/modelscope/trainers/hooks/__init__.py @@ -26,7 +26,7 @@ else: 'iter_timer_hook': ['IterTimerHook'], 'logger': ['TensorboardHook', 'TextLoggerHook'], 'lr_scheduler_hook': ['LrSchedulerHook', 'NoneLrSchedulerHook'], - 'optimizer_hook': [ + 'optimizer': [ 'ApexAMPOptimizerHook', 'NoneOptimizerHook', 'OptimizerHook', 'TorchAMPOptimizerHook' ],