From 6ab85256af546ff2b6e9331d18f9e71dfaa3d52e Mon Sep 17 00:00:00 2001
From: "wenmeng.zwm" <wenmeng.zwm@alibaba-inc.com>
Date: Mon, 26 Jun 2023 21:54:01 +0800
Subject: [PATCH] to [#50529030] fix easycv lr hook error Link:
 https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13069804

---
 modelscope/trainers/hooks/lr_scheduler_hook.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modelscope/trainers/hooks/lr_scheduler_hook.py b/modelscope/trainers/hooks/lr_scheduler_hook.py
index facf5155..acaa7ebf 100644
--- a/modelscope/trainers/hooks/lr_scheduler_hook.py
+++ b/modelscope/trainers/hooks/lr_scheduler_hook.py
@@ -112,6 +112,9 @@ class LrSchedulerHook(Hook):
             self.processor.step(trainer)
 
     def _get_log_lr(self, trainer):
+        # forward compatibility with AddLrLogHook in EasyCV
+        if not hasattr(self, 'processor'):
+            self.processor = LrSchedulerProcessor()
         cur_lr = self.processor.get_current_lr(trainer)
         # only record lr of the first param group
         if isinstance(cur_lr, list):