溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章給大家分享的是有關pytorch中怎么查看網絡參數總量呢的內容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。
netG = Generator()
print('# generator parameters:', sum(param.numel() for param in netG.parameters()))
netD = Discriminator()
print('# discriminator parameters:', sum(param.numel() for param in netD.parameters()))補充：PyTorch查看網絡模型的參數量PARAMS和FLOPS等
在PyTorch中，可以使用torchstat這個庫來查看網絡模型的一些信息，包括總的參數量params、MAdd、顯卡內存占用量和FLOPs等。
from torchstat import stat
from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d
model = resnet50()
stat(model, (3, 224, 224))感謝各位的閱讀！關于“pytorch中怎么查看網絡參數總量呢”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，讓大家可以學到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
