深入研究PyTorch源码:基于AlexNet和Vgg模型的实现与分析
发布时间
阅读量:
阅读量
在Windows操作系统环境下,相关模型的存储路径设定为:C:\Python35\Lib\site-packages\torchvision\models\xxnet.py,该文件中包含了大量常见的网络架构定义,例如AlexNet、ResNet、Inception以及DenseNet等,用户可直接调用这些预定义结构进行应用。
from torchvision import models
model = models.alexnet(pretrained=False)
开始分析AlexNet源代码的结构:
首先需要引入相关的库文件,随后设置 all 变量,该变量的作用是减少对命名空间的干扰。在使用 from xx import * 的方式导入时,如果没有明确指定 all ,则会将当前模块中所有变量一并导出,这可能导致导入方获取到不必要的内容,甚至覆盖其自身定义的变量名。而当模块中定义了 all 列表后,则只会将列表中列出的变量名进行导出。此处与 _X 的用法正好相反:all 用于声明允许导出的变量名,而 _X 则用于标识不应被导出的变量名。Python 在执行 from xx import * 时,会优先查找模块中的 all 列表;若未找到该列表,则默认将所有非以下划
全部评论 (0)
还没有任何评论哟~
