from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d model = resnet50() stat(model, (3, 224, 224)) # (3,224,224)表示輸入圖片的尺寸 使用torchstat這個庫來查看網(wǎng)絡(luò)模型的一些信息,包括總的參數(shù)量params、MAdd、顯卡內(nèi)存占用量和FLOPs等。需要安裝torchstat: 1 pip inst...
www.dbjr.com.cn/article/2137...htm 2025-5-30