Ver código fonte

补充缺失方法

lxylxy123321 1 semana atrás
pai
commit
2dd476b65c
1 arquivos alterados com 4 adições e 1 exclusões
  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):