From 15d00d2b03b88b6a39fbe13614b35b06d122014e Mon Sep 17 00:00:00 2001 From: Xuehai Pan Date: Sat, 19 Nov 2022 22:55:09 +0800 Subject: [PATCH] docs: update api references --- docs/source/api/api.rst | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source/api/api.rst b/docs/source/api/api.rst index 63950e30..27d16a64 100644 --- a/docs/source/api/api.rst +++ b/docs/source/api/api.rst @@ -187,7 +187,6 @@ Optimizer Hooks register_hook zero_nan_hook nan_to_num_hook - nan_to_num Hook ~~~~ @@ -195,23 +194,24 @@ Hook .. autofunction:: register_hook .. autofunction:: zero_nan_hook .. autofunction:: nan_to_num_hook -.. autofunction:: nan_to_num ------ Gradient Transformation ======================= -.. currentmodule:: torchopt.clip +.. currentmodule:: torchopt .. autosummary:: clip_grad_norm + nan_to_num Transforms ~~~~~~~~~~ .. autofunction:: clip_grad_norm +.. autofunction:: nan_to_num Optimizer Schedules ===================