소스 검색

补充缺失方法

lxylxy123321 1 주 전
부모
커밋
2dd476b65c
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      backend/app/engines/text_engine.py

+ 4 - 1
backend/app/engines/text_engine.py

@@ -397,7 +397,10 @@ class _ProgressCallback:
     def on_substep_end(self, args, state, control, logs=None, **kwargs):
         pass
 
-    def on_evaluate(self, args, state, control, metrics=None, **kwargs):
+    def on_pre_optimizer_step(self, args, state, control, logs=None, **kwargs):
+        pass
+
+    def on_prediction_step(self, args, state, control, **kwargs):
         pass
 
     def on_save(self, args, state, control, **kwargs):