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