📌  相关文章
📜  pytorch 汇总模型 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:35.300000             🧑  作者: Mango

代码示例1
from torchsummary import summary
help(summary)
import torchvision.models as models
alexnet = models.alexnet(pretrained=False)
alexnet.cuda()
summary(alexnet, (3, 224, 224))
print(alexnet)