Răsfoiți Sursa

补充缺失方法

lxylxy123321 1 săptămână în urmă
părinte
comite
2dd476b65c
1 a modificat fișierele cu 4 adăugiri și 1 ștergeri
  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):