遍歷named_parameters()中的所有的參數，只打印那些param.requires_grad=True的變量。

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

總結

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

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