From c74aafa87dfb459667917b2b0e73d2a66939b512 Mon Sep 17 00:00:00 2001 From: yukavio <67678385+yukavio@users.noreply.github.com> Date: Tue, 5 Jan 2021 17:03:47 +0800 Subject: [PATCH] fix prune doc (#586) --- docs/zh_cn/api_cn/prune_api.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/zh_cn/api_cn/prune_api.rst b/docs/zh_cn/api_cn/prune_api.rst index 540dae04..e06d16c5 100644 --- a/docs/zh_cn/api_cn/prune_api.rst +++ b/docs/zh_cn/api_cn/prune_api.rst @@ -12,7 +12,7 @@ Pruner **å‚数:** -- **criterion** - 评估一个å·ç§¯å±‚内通é“é‡è¦æ€§æ‰€å‚è€ƒçš„æŒ‡æ ‡ã€‚ç›®å‰æ”¯æŒ ``l1_norm`` , ``bn_scale`` , ``geometry_median`` 。默认为 ``l1_norm`` 。若该å‚数设为 ``bn_scale`` , 则表示剪æžç®—æ³•å°†æ ¹æ®å·ç§¯å±‚åŽè¿žæŽ¥çš„BatchNorm层的Scaleå‚æ•°çš„ç»å¯¹å€¼å¤§å°ä½œä¸ºè¯„ä¼°å·ç§¯å±‚内通é“é‡è¦æ€§æ‰€å‚è€ƒçš„æŒ‡æ ‡ã€‚è‹¥å‚数设为 ``geometry_median``, 则表示剪æžç®—法将基于å·åŸºå±‚内通é“çš„å‡ ä½•ä¸å¿ƒä½œä¸ºè¯„ä¼°å·ç§¯å±‚内通é“é‡è¦æ€§å‚è€ƒæŒ‡æ ‡ã€‚ 在åˆå§‹åŒ–Pruner()ç±»å®žä¾‹æ—¶ï¼Œè‹¥æ²¡æœ‰ä¼ å…¥è¯¥å‚数,则表示Pruner()使用criterion默认å‚数值 ``l1_norm`` ï¼›å¯ä»¥æ˜¾ç¤ºåœ°ä¼ å…¥criterion的值以改å˜å‰ªæžç®—法的剪æžç–略。 +- **criterion** - 评估一个å·ç§¯å±‚内通é“é‡è¦æ€§æ‰€å‚è€ƒçš„æŒ‡æ ‡ã€‚ç›®å‰æ”¯æŒ ``l1_norm`` , ``bn_scale`` , ``geometry_median`` 。默认为 ``l1_norm`` 。若该å‚数设为 ``bn_scale`` , 则表示剪æžç®—æ³•å°†æ ¹æ®å·ç§¯å±‚åŽè¿žæŽ¥çš„BatchNorm层的Scaleå‚æ•°çš„ç»å¯¹å€¼å¤§å°ä½œä¸ºè¯„ä¼°å·ç§¯å±‚内通é“é‡è¦æ€§æ‰€å‚è€ƒçš„æŒ‡æ ‡ã€‚è‹¥å‚数设为 ``geometry_median``, 则表示剪æžç®—法将基于å·ç§¯å±‚内通é“çš„å‡ ä½•ä¸å¿ƒä½œä¸ºè¯„ä¼°å·ç§¯å±‚内通é“é‡è¦æ€§å‚è€ƒæŒ‡æ ‡ã€‚ 在åˆå§‹åŒ–Pruner()ç±»å®žä¾‹æ—¶ï¼Œè‹¥æ²¡æœ‰ä¼ å…¥è¯¥å‚数,则表示Pruner()使用criterion默认å‚数值 ``l1_norm`` ï¼›å¯ä»¥æ˜¾ç¤ºåœ°ä¼ å…¥criterion的值以改å˜å‰ªæžç®—法的剪æžç–略。 - **idx_selector** - 基于å·ç§¯å±‚内通é“é‡è¦æ€§åˆ†æ•°ï¼ŒæŒ‡ç¤ºé€‰æ‹©è£å‰ªçš„å·ç§¯å±‚内通é“索引的ç–略。目å‰æ”¯æŒ ``default_idx_selector`` å’Œ ``optimal_threshold`` 两ç§é€‰æ‹©ç–略。默认为 ``default_idx_selector`` 。 ``default_idx_selector`` ç–ç•¥è¡¨ç¤ºæ ¹æ®å·ç§¯å±‚内通é“çš„é‡è¦æ€§åˆ†æ•°è¿›è¡Œé€‰æ‹©è¦è¢«è£å‰ªçš„通é“。 ``optimal_threshold`` ç–略和 ``bn_scale`` 准则é…åˆä½¿ç”¨ï¼Œå³å°† ``criterion`` 设置为 ``bn_scale`` , 并将该å‚数设置为 ``optimal_threshold``, è¡¨ç¤ºæ ¹æ®å·ç§¯å±‚åŽé“¾æŽ¥çš„BatchNorm层的Scaleå‚数计算出è¦è£å‰ªçš„最优è£å‰ªé˜ˆå€¼ï¼Œå¹¶æ ¹æ®è¯¥é˜ˆå€¼è¿›è¡Œé€šé“è£å‰ªã€‚在åˆå§‹è¯Pruner()å®žä¾‹æ—¶ï¼Œè‹¥æ²¡æœ‰ä¼ å…¥è¯¥å‚数,则表示Pruner()使用idx_selector默认å‚æ•° ``default_idx_selector`` 。 **返回:** 一个Pruner类的实例 -- GitLab