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查看網(wǎng)絡(luò)模型的參數(shù)量PARAMS和FLOPS等
在PyTorch中，可以使用torchstat這個庫來查看網(wǎng)絡(luò)模型的一些信息，包括總的參數(shù)量params、MAdd、顯卡內(nèi)存占用量和FLOPs等。
from torchstat import stat from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d model = resnet50() stat(model, (3, 224, 224))
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
標(biāo)簽：黑龍江 四川 益陽 惠州 鷹潭 黔西 上海 常德
巨人網(wǎng)絡(luò)通訊聲明：本文標(biāo)題《pytorch 一行代碼查看網(wǎng)絡(luò)參數(shù)總量的實現(xiàn)》，本文關(guān)鍵詞 pytorch,一行,代碼,查看,網(wǎng)絡(luò),；如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題，煩請?zhí)峁┫嚓P(guān)信息告之我們，我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò)，涉及言論、版權(quán)與本站無關(guān)。
