Add proper documentation for Keras callbacks (#15374)

* Add proper documentation for Keras callbacks * Add dummies
2022-01-27 10:51:38 -05:00
parent 2de90beeeb
commit 8f6454bfac
4 changed files with 47 additions and 32 deletions
--- a/docs/source/main_classes/keras_callbacks.mdx
+++ b/docs/source/main_classes/keras_callbacks.mdx
@@ -15,6 +15,10 @@ specific language governing permissions and limitations under the License.
 When training a Transformers model with Keras, there are some library-specific callbacks available to automate common
 tasks:

+## KerasMetricCallback
+
+[[autodoc]] KerasMetricCallback
+
 ## PushToHubCallback

-[[autodoc]] keras_callbacks.PushToHubCallback
+[[autodoc]] PushToHubCallback