當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python mxnet.gluon.nn.Block.collect_params用法及代碼示例


用法:

collect_params(select=None)

參數：

select：(str) - 常用表達

返回：

返回類型：

選中的ParameterDict

返回包含此 Block 及其所有子參數(默認)的 ParameterDict，也可以返回匹配某些給定正則表達式的選擇 ParameterDict。

例如收集[‘conv1_weight’, ‘conv1_bias’,‘fc_weight’, ‘fc_bias’]中的指定參數：

model.collect_params('conv1_weight|conv1_bias|fc_weight|fc_bias')

或收集名稱以‘weight’ or ‘bias’ 結尾的所有參數，這可以使用正則表達式完成：

model.collect_params('.*weight|.*bias')

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.gluon.nn.Block.collect_params。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。