pytorch查看模型的参数总量、占用显存量以及flops
·
from torchstat import stat
import torchvision.models as models
model = models.resnet152()
stat(model, (3, 224, 224))

更多推荐
from torchstat import stat
import torchvision.models as models
model = models.resnet152()
stat(model, (3, 224, 224))

更多推荐
所有评论(0)