PyTorch和Keras計算模型參數(shù)的例子
更新時間：2020年01月02日 17:26:03 作者：咆哮的阿杰
今天小編就為大家分享一篇PyTorch和Keras計算模型參數(shù)的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
Pytorch中，變量參數(shù)，用numel得到參數(shù)數(shù)目，累加
def get_parameter_number(net):
total_num = sum(p.numel() for p in net.parameters())
trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
return {'Total': total_num, 'Trainable': trainable_num}
Keras中，直接使用model的summary函數(shù)
model = k_model() model.summary()
以上這篇PyTorch和Keras計算模型參數(shù)的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了python實現(xiàn)隨機密碼字典生成器示例,需要的朋友可以參考下2014-04-04
這篇文章主要為大家介紹了Python進階之協(xié)程，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
這篇文章主要介紹了在Django的View中使用asyncio的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-07-07
這篇文章主要介紹了python入門-jupyter基礎操作及文本用法，適合入門的python小白閱讀，有需要的朋友可以借鑒參考下，希望可以有所幫助，感謝閱讀2021-09-09
這篇文章主要介紹了Python字符串中如何去除數(shù)字之間的逗號，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
這篇文章主要介紹了Python實現(xiàn)打印http請求信息方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-06-06
如果Anaconda的Jupyter Notebook無法打開并且頻繁閃退,可能是由于一些配置問題或者軟件沖突引起的,本文就來介紹一下jupyter閃退的解決方法及卸載與安裝教程,感興趣的可以了解一下2023-11-11
numpy判斷數(shù)值類型、過濾出數(shù)值型數(shù)據(jù)的方法
今天小編就為大家分享一篇numpy判斷數(shù)值類型、過濾出數(shù)值型數(shù)據(jù)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
詳解python 拆包可迭代數(shù)據(jù)如tuple, list
拆包是指將一個結構中的數(shù)據(jù)拆分為多個單獨變量中。下面通過本文給大家介紹python 拆包可迭代數(shù)據(jù)如tuple, list的相關資料，需要的朋友參考下吧2017-12-12
Python多線程同步Lock、RLock、Semaphore、Event實例
這篇文章主要介紹了Python多線程同步Lock、RLock、Semaphore、Event實例,Lock & RLock 用來確保多線程多共享資源的訪問,Semaphore用來確保一定資源多線程訪問時的上限,Event是最簡單的線程間通信的方式,需要的朋友可以參考下2014-11-11