浏览代码

补充缺失方法

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):