predict.luz_module_fitted callbacks help
jonthegeek opened this issue · comments
luz_callback lists the training breakpoints at which callbacks can be called, but not when they can be called during predict.luz_module_fitted. We should add that information to luz_callback help.
Digging through the code, these are what I see:
- on_predict_begin
- on_predict_batch_begin
- on_predict_batch_end
- on_predict_end
Did I miss any?