深度学习笔记-快速查看torchvision中的模型架构

在模型创建的时候,经常需要对模型进行修改,修改的时候需要查看模型架构。下面使用终端中调用IPython的方法对torchvision中的模型架构进行展示

查询代码:

首先ipython激活ipython

from torchvision import models

print(models.resnet152())

深度学习笔记-快速查看torchvision中的模型架构_第1张图片

  • 注意:resnet152后面要加()

你可能感兴趣的:(深度学习笔记,深度学习,架构,人工智能)