查看PyTorch自带的模型的结构

from __future__ import print_function
import torch,torchvision

net = torchvision.models.densenet121()
print(net)
posted @ 2017-12-26 16:47  菜鸡一枚  阅读(2912)  评论(0)    收藏  举报