Pytorch 統計模型參數量

• • param.numel()
  • 統計模型參數量

param.numel()

返回param中元素的數量

統計模型參數量

num_params = sum(param.numel() for param in net.parameters()) print(num_params)

總結

以上是生活随笔為你收集整理的Pytorch 统计模型参数量 param.numel()的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。