for name, param in model.named_parameters():if param.requires_grad:print(name,':',param.size())

?

總結

以上是生活随笔為你收集整理的pytorch查看模型的可训练参数的全部內容，希望文章能夠幫你解決所遇到的問題。

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