PyTorch构造神经网络及各modules解析

xiangyong58

已于 2022-11-18 17:06:59 修改

阅读量479

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Machine & Deep Learning 文章标签： pytorch 神经网络

于 2020-01-02 16:52:34 首次发布

本文链接：https://blog.youkuaiyun.com/xiangyong58/article/details/103807279

Machine & Deep Learning 专栏收录该内容

78 篇文章 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

本文详细介绍了如何使用PyTorch构建神经网络，包括使用torchvision.models构建常见基线网络，SENet模块解析，ResNet结构梳理，以及多GPU使用等。此外，还探讨了网络创建、预训练模型加载、评估过程中的常见误区。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

零：用torchvision.models构造常用baselines网络

import torchvision.models as models

pytorch已定义models详看：Models and pre-trained weights — Torchvision main documentation

构造示例：

import torchvision.models as models


alexnet = models.alexnet()                         # AlexNet
vgg16 = models.vgg16()                             # VGG16
resnet18 = models.resnet18()                       # ResetNet模型
googlenet = models.googlenet()                     # googlenet
inception = models.inception_v3()                  # inception

squeezenet = models.squeezenet1_0()                #